nroggendorff commited on
Commit
9d21348
Β·
verified Β·
1 Parent(s): 1bbb208

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import torch
5
  from diffusers import StableDiffusionXLPipeline
6
 
7
  pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0").to("cuda")
 
8
 
9
  @spaces.GPU
10
  def generate(prompt, negative_prompt, width, height, sample_steps):
 
5
  from diffusers import StableDiffusionXLPipeline
6
 
7
  pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0").to("cuda")
8
+ pipeline.load_lora_weights("nroggendorff/zelda-lora")
9
 
10
  @spaces.GPU
11
  def generate(prompt, negative_prompt, width, height, sample_steps):