Downloads · 30 days
0
ybelkada/sd-1.5-pokemon-lora-peft
sd-1.5-pokemon-lora-peft is a machine learning model from ybelkada. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
For now, use diffusers and peft main then:
Downloads · 30 days
0
Access
Public
Updated Oct 13, 2023
Repo size
27.7 MB
Likes
0
Public
Click a slice to open those files.
.safetensors9.2 MB · 100%
From the Hugging Face model README
For now, use diffusers and peft main then:
import torch
from diffusers import DiffusionPipeline
model_id = "runwayml/stable-diffusion-v1-5"
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(1)
pipe.load_lora_weights("/home/younes_huggingface_co/code/diffusers/sd-model-finetuned-lora/checkpoint-1200")
prompt = "cute dragon creature"
image = pipe(prompt).images[0]
image.save("my_image.png")