Update README.md

README.md

@@ -8,6 +8,32 @@ language:
 tags:
 - completion
 ---
+## Info
+
+Magic prompt completion model trained on a dataset 70k Stable Diffusion prompts. Base model: TinyStories-33M.
+
+Model seems to be pretty decent for 33M params, but it clearly lacks much of an understanding of pretty much anything. Still, considering the size, I think it's decent. Whether you would use this over a small GPT-2 based model is up to you.
+
+## Examples
+
+Generation settings: `max_new_tokens=40, do_sample=True, temperature=2.0, num_beams=10, repetition_penalty=1.2, top_k=40, top_p=0.75, eos_token_id=tokenizer.eos_token_id` (there may be better settings).
+
+(Bold text is generated by the model)
+
+"A close shot of a bird in a jungle, **with two legs, with long hair on a tall, long brown body, long white skin, sharp teeth, high bones, digital painting, artstation, concept art, illustration by wlop,**"
+
+"Camera shot of **a strange young girl wearing a cloak, wearing a mask in clothes, with long curly hair, long hair, black eyes, dark skin, white teeth, long brown eyes eyes, big eyes, sharp**"
+
+"An illustration of a house, stormy weather, **sun, moonlight, night, concept art, 4 k, wlop, by wlop, by jose stanley, ilya kuvshinov, sprig**"
+
+"A field of flowers, camera shot, 70mm lens, **fantasy, intricate, highly detailed, artstation, concept art, sharp focus, illustration, illustration, artgerm jake daggaws, artgerm and jaggodieie brad**"
+
+## Training config
+
+- Rank 16 LoRA
+- Trained on Gustavosta/Stable-Diffusion-Prompts for 10 epochs
+- Batch size of 64
+
 ## Training procedure