Update README.md
Browse files
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)
|