Hugging Face's logo

RWKV
/
rwkv-4-world-1b5

Transformers
PyTorch
rwkv
Inference Endpoints
Model card Files Community
1
KaleiNeely commited on
Commit
9f2aa8f
•
1 Parent(s): bf8d61d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -82,7 +82,7 @@ User: {instruction}
82
  Assistant:"""
83
 
84
 
85
- model = AutoModelForCausalLM.from_pretrained("RWKV/rwkv-4-world-1b5", trust_remote_code=True, torch_dtype=torch.float16).to(0)
86
  tokenizer = AutoTokenizer.from_pretrained("RWKV/rwkv-4-world-1b5", trust_remote_code=True)
87
 
88
  text = "乌兰察布"
 
82
  Assistant:"""
83
 
84
 
85
+ model = AutoModelForCausalLM.from_pretrained("RWKV/rwkv-4-world-1b5", trust_remote_code=True, torch_dtype=torch.float32).to(0)
86
  tokenizer = AutoTokenizer.from_pretrained("RWKV/rwkv-4-world-1b5", trust_remote_code=True)
87
 
88
  text = "乌兰察布"