nsfwalex commited on
Commit
1b73eca
1 Parent(s): 981c266

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. inference_manager.py +1 -0
inference_manager.py CHANGED
@@ -502,6 +502,7 @@ class ModelManager:
502
  if not images:
503
  raise Exception(f"face images not provided")
504
  start = time.time()
 
505
  ip_model.pipe.to("cuda")
506
  ip_model.image_proj_model.to("cuda")
507
  if not self.app:
 
502
  if not images:
503
  raise Exception(f"face images not provided")
504
  start = time.time()
505
+ ip_model.pipe.sampler = samplers["Euler a"]
506
  ip_model.pipe.to("cuda")
507
  ip_model.image_proj_model.to("cuda")
508
  if not self.app: