Downloads · 30 days
9
1% of all-time downloads
sd-pokemon-model-lora
sr5434/sd-pokemon-model-lora
sd-pokemon-model-lora is a text-to-image model from sr5434. Use it when you need an image from a text prompt. It is set up for diffusers.
I trained this model using LoRA and Stable Diffusion v1.4
All-time downloads
1.3K
Public
Repo size
3.3 MB
Likes
3
Public
Repo makeup
Click a slice to open those files.
.bin6.6 MB · 100%
At a glance
- Task
- Text-to-Image
- Library
- diffusers
- Access
- Public
- Created
- Jan 30, 2023
- Updated
- Feb 25, 2023
- SHA
- 2ef486fd
Datasets
Model card
From the Hugging Face model README
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")
- Task
- Text-to-Image
- Library
- diffusers
- Languages
- en
- Created
- Jan 30, 2023
- Updated
- Feb 25, 2023