MinaGabriel commited on
Commit
b7df38e
·
verified ·
1 Parent(s): 76f30ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -39,7 +39,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
39
  from peft import PeftModel
40
 
41
  BASE = "microsoft/phi-2"
42
- ADAPTER = "MinaGabriel/phi2-lora-nars"
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
44
 
45
  tok = AutoTokenizer.from_pretrained(BASE, use_fast=True)
 
39
  from peft import PeftModel
40
 
41
  BASE = "microsoft/phi-2"
42
+ ADAPTER = "MinaGabriel/phi2-2.7b-lora-nars-adapter"
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
44
 
45
  tok = AutoTokenizer.from_pretrained(BASE, use_fast=True)