Minami-su
/

Qwen1.5-14B-Chat_mistral

Text Generation

text-generation-inference

Model card Files Files and versions Community

Minami-su commited on Feb 25, 2024

Commit

1473b9c

·

verified ·

1 Parent(s): 00dfe39

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,8 +46,8 @@ Usage:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
-tokenizer = AutoTokenizer.from_pretrained("Minami-su/Qwen1.5-7B-Chat_mistral")
-model = AutoModelForCausalLM.from_pretrained("Minami-su/Qwen1.5-7B-Chat_mistral", torch_dtype="auto", device_map="auto")
 streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 messages = [

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+tokenizer = AutoTokenizer.from_pretrained("Minami-su/Qwen1.5-14B-Chat_mistral")
+model = AutoModelForCausalLM.from_pretrained("Minami-su/Qwen1.5-14B-Chat_mistral", torch_dtype="auto", device_map="auto")
 streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 messages = [