Update README.md
Browse files
README.md
CHANGED
|
@@ -51,7 +51,7 @@ tokenizer = AutoTokenizer.from_pretrained("cyberagent/calm2-7b-chat")
|
|
| 51 |
tokenizer.pad_token = tokenizer.eos_token
|
| 52 |
|
| 53 |
model = RetNetForCausalLM.from_pretrained(
|
| 54 |
-
"Spiral-AI/RetNet-3b-base
|
| 55 |
)
|
| 56 |
inputs = tokenizer("最近、秋葉原周辺で興味深い", return_tensors="pt")
|
| 57 |
input_ids = inputs["input_ids"].to(model.device)
|
|
|
|
| 51 |
tokenizer.pad_token = tokenizer.eos_token
|
| 52 |
|
| 53 |
model = RetNetForCausalLM.from_pretrained(
|
| 54 |
+
"Spiral-AI/Spiral-RetNet-3b-base", device_map="auto"
|
| 55 |
)
|
| 56 |
inputs = tokenizer("最近、秋葉原周辺で興味深い", return_tensors="pt")
|
| 57 |
input_ids = inputs["input_ids"].to(model.device)
|