AlekseyCalvin commited on
Commit
2bcefbc
·
verified ·
1 Parent(s): dd188e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -112,7 +112,7 @@ def update_selection(evt: gr.SelectData, width, height):
112
  )
113
 
114
  @spaces.GPU(duration=70)
115
- def generate_image(prompt, negative_prompt, lora_scale, steps, seed, cfg_scale, width, height):
116
  pipe.to("cuda")
117
  generator = torch.Generator(device="cuda").manual_seed(seed)
118
 
 
112
  )
113
 
114
  @spaces.GPU(duration=70)
115
+ def generate_image(prompt, negative_prompt, lora_scale=1.0, steps, seed, cfg_scale, width, height):
116
  pipe.to("cuda")
117
  generator = torch.Generator(device="cuda").manual_seed(seed)
118