Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Update app.py
Browse files
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,
|
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, width, height, steps, seed, lora_scale=1.0, cfg_scale=3.5):
|
116 |
pipe.to("cuda")
|
117 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
118 |
|