Spaces:
Runtime error
Runtime error
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)
|