Spaces:
Runtime error
Runtime error
| 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, | |
| cross_attention_kwargs={"scale": 0.5}, | |
| ).images[0] | |
| image.save("test1.png") |