vicky4s4s
/

Qwen2-1.5B

Text Generation

text-generation-inference

Model card Files Files and versions

vicky4s4s commited on Jun 12, 2024

Commit

e8482a0

·

verified ·

1 Parent(s): 05f93c3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
-    "Qwen/Qwen2-1.5B-Instruct",
     torch_dtype="auto",
     device_map="auto"
 )

 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
+    "vicky4s4s/Qwen2-1.5B",
     torch_dtype="auto",
     device_map="auto"
 )