Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ class calculateDuration:
|
|
57 |
|
58 |
@spaces.GPU(duration=120)
|
59 |
@torch.inference_mode()
|
60 |
-
def generate_image(prompt, lora_strings_json,
|
61 |
|
62 |
lora_configs = None
|
63 |
adapter_names = []
|
@@ -178,7 +178,7 @@ def run_lora(prompt, lora_strings_json, cfg_scale, steps, randomize_seed, seed,
|
|
178 |
error_message = ""
|
179 |
try:
|
180 |
print("Start applying for zeroGPU resources")
|
181 |
-
final_image = generate_image(prompt, lora_strings_json,
|
182 |
except Exception as e:
|
183 |
error_message = str(e)
|
184 |
gr.Error(error_message)
|
|
|
57 |
|
58 |
@spaces.GPU(duration=120)
|
59 |
@torch.inference_mode()
|
60 |
+
def generate_image(prompt, lora_strings_json, steps, seed, cfg_scale, width, height, progress):
|
61 |
|
62 |
lora_configs = None
|
63 |
adapter_names = []
|
|
|
178 |
error_message = ""
|
179 |
try:
|
180 |
print("Start applying for zeroGPU resources")
|
181 |
+
final_image = generate_image(prompt, lora_strings_json, steps, seed, cfg_scale, width, height, progress)
|
182 |
except Exception as e:
|
183 |
error_message = str(e)
|
184 |
gr.Error(error_message)
|