RWKV-Red-Team
/

ARWKV-7B-Preview-0.1-NoG

Text Generation

Model card Files Files and versions Community

zhiyuan8 commited on 14 days ago

Commit

5b28c59

·

verified ·

1 Parent(s): b1f3dab

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -54,13 +54,13 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model = AutoModelForCausalLM.from_pretrained(
-    "RWKV-Red-Team/ARWKV-7B-Preview-0.1",
     device_map="auto",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 )
 tokenizer = AutoTokenizer.from_pretrained(
-    "RWKV-Red-Team/ARWKV-7B-Preview-0.1"
 )
 ```

 model = AutoModelForCausalLM.from_pretrained(
+    "RWKV-Red-Team/ARWKV-7B-Preview-0.1-NoG",
     device_map="auto",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 )
 tokenizer = AutoTokenizer.from_pretrained(
+    "RWKV-Red-Team/ARWKV-7B-Preview-0.1-NoG"
 )
 ```