Hugging Face's logo

BBuf
/
RWKV-4-World-169M

Transformers
PyTorch
rwkv
Inference Endpoints
Model card Files Community
BBuf commited on
Commit
50f9fc5
•
1 Parent(s): 714d198

refine

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ Answer: The researchers discovered a mysterious finding in a remote, undisclosed
31
  import torch
32
  from transformers import AutoModelForCausalLM, AutoTokenizer
33
 
34
- model = AutoModelForCausalLM.from_pretrained("BBuf/RWKV-4-World-169M", torch_dtype=torch.float16).to(0)
35
  tokenizer = AutoTokenizer.from_pretrained("BBuf/RWKV-4-World-169M", trust_remote_code=True)
36
 
37
  text = "你叫什么名字？"
 
31
  import torch
32
  from transformers import AutoModelForCausalLM, AutoTokenizer
33
 
34
+ model = AutoModelForCausalLM.from_pretrained("BBuf/RWKV-4-World-169M", torch_dtype=torch.float32).to(0)
35
  tokenizer = AutoTokenizer.from_pretrained("BBuf/RWKV-4-World-169M", trust_remote_code=True)
36
 
37
  text = "你叫什么名字？"