Create README.md

README.md

@@ -0,0 +1,31 @@
+---
+datasets:
+- lambdalabs/pokemon-blip-captions
+language:
+- en
+library_name: diffusers
+pipeline_tag: text-to-image
+---
+
+# Model Card for my pokemon generating AI
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+I trained this model using LoRA and Stable Diffusion v1.4
+# Model Details
+
+## How to use
+Sample code:
+```
+from diffusers import StableDiffusionPipeline
+import torch
+
+model_path = "sr5434/sd-pokemon-model-lora"
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
+pipe.unet.load_attn_procs(model_path)
+pipe.to("cuda")
+
+prompt = input("Prompt:")
+image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
+image.save("pokemon.png")
+```