bernardo-de-almeida commited on
Commit
82970d1
·
1 Parent(s): e2bc640

feat: improve timings

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -496,10 +496,10 @@ def predict(
496
  # only move if needed
497
  if current != device:
498
  pipe.model.to(device)
499
- tprint(f"model.ensure_on({device})")
500
 
501
  pipe.model.eval()
502
- print(f"Running on {next(pipe.model.parameters()).device}")
503
  tprint("model ready to run inference")
504
 
505
  # run inference
 
496
  # only move if needed
497
  if current != device:
498
  pipe.model.to(device)
499
+ tprint(f"model moved to {device}")
500
 
501
  pipe.model.eval()
502
+ tprint(f"Running on {next(pipe.model.parameters()).device}")
503
  tprint("model ready to run inference")
504
 
505
  # run inference