Rustamshry commited on
Commit
b658f3b
·
verified ·
1 Parent(s): 7d8b132

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -59,7 +59,7 @@ login(token="")
59
  tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen3-1.7B",)
60
  base_model = AutoModelForCausalLM.from_pretrained(
61
  "unsloth/Qwen3-1.7B",
62
- device_map="auto", token=""
63
  )
64
 
65
  model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen3-1.7B-azerbaijani-math")
 
59
  tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen3-1.7B",)
60
  base_model = AutoModelForCausalLM.from_pretrained(
61
  "unsloth/Qwen3-1.7B",
62
+ device_map={"": 0}, token=""
63
  )
64
 
65
  model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen3-1.7B-azerbaijani-math")