Update README.md

README.md

@@ -14,17 +14,54 @@ This model is a pretrained RetNet model trained from scratch using https://githu
 It achieves the following results on the evaluation set:
 - Loss: 0.5923
 
+## Usage
+
+```bash
+pip install transformers safetensors
+```
+
+```py
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+
+MODEL_NAME = "isek-ai/SDPrompt-RetNet-v2"
+DEVICE = "cuda"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model= AutoModelForCausalLM.from_pretrained(
+    MODEL_NAME,
+    torch_dtype=torch.float16, # or torch.bfloat16
+    trust_remote_code=True,
+).to(DEVICE)
+model.eval()
+streamer = TextStreamer(tokenizer)
+
+prompt = "1girl"
+inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+_ = model.generate(
+    inputs["input_ids"],
+    max_new_tokens=256,
+    do_sample=True,
+    top_p=0.9,
+    top_k=20,
+    temperature=0.9,
+    streamer=streamer,
+)
+# 1girl, :<, bag, black hair, blurry, bokeh, cloud, depth of field, from side, long sleeves, night, outdoors, pleated skirt, power lines, purple eyes, road, scenery, shoes, shoulder bag,gasm, sidelocks, sign, skirt,let's drawsaurus, skylight smile, sneakers, standing, star (sky), sweater, town, traffic cone, utility pole, vending machine, wide-eyed, window, wooden box, yellow skirt,ization, zettai ryouiki, zoom layer, white footwear, zipper, zipper pull tab, zipperland sheet, zombie pose, ladder, leaning back, leg up, looking to the side,let, miniskirt, motion blur, musical note, open mouth, part
+```
+
+
 ## Model description
 
-More information needed
+This model is trained with *:only Danbooru tags** to generate prompts for image generation models.
 
-## Intended uses & limitations
+## Training data
 
-More information needed
+- [isek-ai/danbooru-tags-2016-2023](https://huggingface.co/datasets/isek-ai/danbooru-tags-2016-2023)
 
-## Training and evaluation data
+### Dataset filtering
 
-More information needed
+TODO
 
 ## Training procedure