burtenshaw HF Staff commited on
Commit
4206cc4
·
verified ·
1 Parent(s): deb7026

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -116,7 +116,7 @@ import torch
116
  from transformers import AutoModelForCausalLM, AutoTokenizer
117
 
118
 
119
- model_id="nanochat-students/d20-chat-transformers"
120
  max_new_tokens=64
121
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
122
 
 
116
  from transformers import AutoModelForCausalLM, AutoTokenizer
117
 
118
 
119
+ model_id="nanochat-students/chat-d20"
120
  max_new_tokens=64
121
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
122