isek-ai
/

LightNovel-Intro-RetNet-400M

Text Generation

Generated from Trainer

Model card Files Files and versions Community

p1atdev commited on Oct 21, 2023

Commit

f814b50

•

1 Parent(s): 44ef3fb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a RetNet model trained from scratch using https://github.com/syncdoth/RetNet.
-Demo: https://huggingface.co/spaces/p1atdev/LightNovel-Intro-RetNet-400M-Demo
 ## Usage
@@ -38,7 +38,7 @@ Example inference script:
 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
-MODEL_NAME = "p1atdev/LightNovel-Intro-RetNet-400M"
 device = "cuda"

 This model is a RetNet model trained from scratch using https://github.com/syncdoth/RetNet.
+Demo: https://huggingface.co/spaces/isek-ai/LightNovel-Intro-RetNet-400M-Demo
 ## Usage
 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
+MODEL_NAME = "isek-ai/LightNovel-Intro-RetNet-400M"
 device = "cuda"