Qwen
/

Qwen1.5-1.8B-Chat

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

JustinLin610 commited on Feb 2

Commit

31c0e98

•

1 Parent(s): cbdbf5f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -47,10 +47,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
-    "Qwen2/Qwen2-beta-1_8B-Chat",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("Qwen2/Qwen2-beta-1_8B-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-1_8B-Chat",
     device_map="auto"
 )
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-beta-1_8B-Chat")
 prompt = "Give me a short introduction to large language model."
 messages = [