Spaces:
Running
on
T4
Running
on
T4
File size: 477 Bytes
f4ff201 74ce001 f4ff201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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") |