Update README.md
Browse files
README.md
CHANGED
@@ -63,7 +63,7 @@ login(token="")
|
|
63 |
tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen3-0.6B",)
|
64 |
base_model = AutoModelForCausalLM.from_pretrained(
|
65 |
"unsloth/Qwen3-0.6B",
|
66 |
-
device_map="
|
67 |
)
|
68 |
|
69 |
model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen3-0.6B-OpenMathReason")
|
|
|
63 |
tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen3-0.6B",)
|
64 |
base_model = AutoModelForCausalLM.from_pretrained(
|
65 |
"unsloth/Qwen3-0.6B",
|
66 |
+
device_map={"": 0}, token=""
|
67 |
)
|
68 |
|
69 |
model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen3-0.6B-OpenMathReason")
|