Spaces:
Running
on
Zero
Running
on
Zero
Update app_quant.py
Browse files- app_quant.py +1 -1
app_quant.py
CHANGED
|
@@ -229,7 +229,7 @@ log_system_stats("AFTER PIPELINE BUILD")
|
|
| 229 |
# ============================================================
|
| 230 |
# INFERENCE
|
| 231 |
# ============================================================
|
| 232 |
-
@spaces.
|
| 233 |
def generate_image(prompt, height, width, steps, seed):
|
| 234 |
global LOGS
|
| 235 |
LOGS = "" # reset logs
|
|
|
|
| 229 |
# ============================================================
|
| 230 |
# INFERENCE
|
| 231 |
# ============================================================
|
| 232 |
+
@spaces.GPU
|
| 233 |
def generate_image(prompt, height, width, steps, seed):
|
| 234 |
global LOGS
|
| 235 |
LOGS = "" # reset logs
|