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")