Qwen
/

Qwen1.5-14B-Chat

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

JustinLin610 commited on Jan 31

Commit

c56a582

•

1 Parent(s): f7cc81f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -48,8 +48,8 @@ Here provides a code snippet with `apply_chat_template` to show you how to load
 from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
-model = AutoModelForCausalLM.from_pretrained("Qwen2/Qwen2-beta-7B-Chat", device_map="auto")
-tokenizer = AutoTokenizer.from_pretrained("Qwen2/Qwen2-beta-7B-Chat")
 prompt = "Give me a short introduction to large language model."

 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."