Rustamshry commited on
Commit
7417fc8
verified
1 Parent(s): 0ca6063

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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="auto", token=""
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")