Spaces:
Running
on
T4
Running
on
T4
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 |
|