raincandy-u commited on
Commit
ae4a4fa
1 Parent(s): 110095b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -62,8 +62,8 @@ from transformers import AutoTokenizer
62
  import transformers
63
  import torch
64
 
65
- model = "mlabonne/AlphaMonarch-7B"
66
- messages = [{"role": "user", "content": "What is a large language model?"}]
67
 
68
  tokenizer = AutoTokenizer.from_pretrained(model)
69
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
62
  import transformers
63
  import torch
64
 
65
+ model = "raincandy-u/Rain-7B-v0.1"
66
+ messages = [{"role": "user", "content": "What is chain of thoughts?"}]
67
 
68
  tokenizer = AutoTokenizer.from_pretrained(model)
69
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)