Update README.md
Browse files
README.md
CHANGED
|
@@ -57,9 +57,9 @@ from peft import PeftModel
|
|
| 57 |
|
| 58 |
login(token="")
|
| 59 |
|
| 60 |
-
tokenizer = AutoTokenizer.from_pretrained("unsloth/
|
| 61 |
base_model = AutoModelForCausalLM.from_pretrained(
|
| 62 |
-
"unsloth/
|
| 63 |
device_map={"": 0}, token=""
|
| 64 |
)
|
| 65 |
|
|
|
|
| 57 |
|
| 58 |
login(token="")
|
| 59 |
|
| 60 |
+
tokenizer = AutoTokenizer.from_pretrained("unsloth/Llama-3.2-1B-Instruct",)
|
| 61 |
base_model = AutoModelForCausalLM.from_pretrained(
|
| 62 |
+
"unsloth/Llama-3.2-1B-Instruct",
|
| 63 |
device_map={"": 0}, token=""
|
| 64 |
)
|
| 65 |
|