Krebzonide's picture
Update app.py
74ce001
raw
history blame
477 Bytes
import torch
from diffusers import StableDiffusionPipeline
model_base = "runwayml/stable-diffusion-v1-5"
lora_model_path = "Krebzonide/3a0s-w68r-4qw1-0"
pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
pipe.unet.load_attn_procs(lora_model_path)
pipe.to("cuda")
image = pipe(
"A picture of gfemti in a black dress, outdoors",
num_inference_steps=25,
guidance_scale=7.5
).images[0]
image.save("test1.png")