Spiral-AI
/

Spiral-RetNet-3b-base

Text Generation

Inference Endpoints

Model card Files Files and versions Community

ksterx commited on May 1

Commit

9cae16e

•

1 Parent(s): 6acd15c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ tokenizer = AutoTokenizer.from_pretrained("cyberagent/calm2-7b-chat")
 tokenizer.pad_token = tokenizer.eos_token
 model = RetNetForCausalLM.from_pretrained(
-    "Spiral-AI/RetNet-3b-base-ja", device_map="auto"
 )
 inputs = tokenizer("最近、秋葉原周辺で興味深い", return_tensors="pt")
 input_ids = inputs["input_ids"].to(model.device)

 tokenizer.pad_token = tokenizer.eos_token
 model = RetNetForCausalLM.from_pretrained(
+    "Spiral-AI/Spiral-RetNet-3b-base", device_map="auto"
 )
 inputs = tokenizer("最近、秋葉原周辺で興味深い", return_tensors="pt")
 input_ids = inputs["input_ids"].to(model.device)