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