Spaces:
Running
on
T4
Running
on
T4
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") |