isek-ai
/

SDPrompt-RetNet-300M

Text Generation

Generated from Trainer

Model card Files Files and versions

p1atdev commited on Nov 8, 2023

Commit

0fbda94

·

1 Parent(s): 2339d5d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ pip install transformers safetensors timm
 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
-MODEL_NAME = ""
 DEVICE = "cuda"

 ```py
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+MODEL_NAME = "isek-ai/SDPrompt-RetNet-300M"
 DEVICE = "cuda"