nyanko7 commited on
Commit
25d37d8
1 Parent(s): 92ee4a3

fix: minor fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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