Qwen
/

Qwen1.5-14B-Chat

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

JustinLin610 commited on Feb 2

Commit

39be9a6

•

1 Parent(s): 47ec9ba

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -49,10 +49,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
-    "Qwen2/Qwen2-beta-14B-Chat",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("Qwen2/Qwen2-beta-14B-Chat")
 prompt = "Give me a short introduction to large language model."
 messages = [

 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
+    "Qwen/Qwen2-beta-14B-Chat",
     device_map="auto"
 )
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-beta-14B-Chat")
 prompt = "Give me a short introduction to large language model."
 messages = [