Spaces:
Runtime error
Runtime error
fix: minor fix
Browse files
app.py
CHANGED
|
@@ -295,7 +295,7 @@ def inference(
|
|
| 295 |
|
| 296 |
end_time = time.time()
|
| 297 |
vram_free, vram_total = torch.cuda.mem_get_info()
|
| 298 |
-
print(f"done: res={width}x{height}, step={steps}, time={round(end_time-start_time, 2)}s, vram_alloc={convert_size(vram_total-vram_free)}/{convert_size(vram_total)}")
|
| 299 |
return gr.Image.update(result[0][0], label=f"Initial Seed: {seed}")
|
| 300 |
|
| 301 |
|
|
|
|
| 295 |
|
| 296 |
end_time = time.time()
|
| 297 |
vram_free, vram_total = torch.cuda.mem_get_info()
|
| 298 |
+
print(f"done: model={model}, res={width}x{height}, step={steps}, time={round(end_time-start_time, 2)}s, vram_alloc={convert_size(vram_total-vram_free)}/{convert_size(vram_total)}")
|
| 299 |
return gr.Image.update(result[0][0], label=f"Initial Seed: {seed}")
|
| 300 |
|
| 301 |
|