Update inference_manager.py
Browse files- 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:
|