isek-ai
/

SDPrompt-RetNet-v2-beta

Text Generation

Model card Files Files and versions Community

p1atdev commited on Dec 26, 2023

Commit

7f0a15e

•

1 Parent(s): b6e68c2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ pip install transformers safetensors
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
-MODEL_NAME = "isek-ai/SDPrompt-RetNet-v2"
 DEVICE = "cuda"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+MODEL_NAME = "isek-ai/SDPrompt-RetNet-v2-beta"
 DEVICE = "cuda"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)