Rustamshry
/

Qwen2.5-1.5B-Medical-Az

Question Answering

Model card Files Files and versions Community

Rustamshry commited on May 20

Commit

3643634

·

verified ·

1 Parent(s): c192c50

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ login(token="")
 tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen2.5-1.5B-Instruct",)
 base_model = AutoModelForCausalLM.from_pretrained(
     "unsloth/Qwen2.5-1.5B-Instruct",
-    device_map="auto", token=""
 )
 model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen2.5-1.5B-Medical-Az")

 tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen2.5-1.5B-Instruct",)
 base_model = AutoModelForCausalLM.from_pretrained(
     "unsloth/Qwen2.5-1.5B-Instruct",
+    device_map={"": 0}, token=""
 )
 model = PeftModel.from_pretrained(base_model,"Rustamshry/Qwen2.5-1.5B-Medical-Az")