Views
No views yet
1import torch
2from diffusers import DiffusionPipeline
3
4model_id = "runwayml/stable-diffusion-v1-5"
5
6pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(1)
7
8pipe.load_lora_weights("/home/younes_huggingface_co/code/diffusers/sd-model-finetuned-lora/checkpoint-1200")
9
10prompt = "cute dragon creature"
11image = pipe(prompt).images[0]
12image.save("my_image.png")