Minami-su
/

Qwen1.5-0.5B-Chat_llamafy

Text Generation

text-generation-inference

Model card Files Files and versions Community

Minami-su commited on Feb 24, 2024

Commit

f67dba6

•

1 Parent(s): 4febfcc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ messages = [
 inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
 inputs = inputs.to("cuda")
 generate_ids = model.generate(inputs,max_length=2048, streamer=streamer)
 ## Test

 inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
 inputs = inputs.to("cuda")
 generate_ids = model.generate(inputs,max_length=2048, streamer=streamer)
+```
 ## Test