Compare commits
1 Commits
8b12692cfe
...
cd1e29c1b1
| Author | SHA1 | Date |
|---|---|---|
|
|
cd1e29c1b1 |
|
|
@ -63,6 +63,8 @@ def get_face_swapper() -> Any:
|
|||
with THREAD_LOCK:
|
||||
if FACE_SWAPPER is None:
|
||||
model_name = "inswapper_128.onnx"
|
||||
if "CUDAExecutionProvider" in modules.globals.execution_providers:
|
||||
model_name = "inswapper_128.onnx"
|
||||
model_path = os.path.join(models_dir, model_name)
|
||||
FACE_SWAPPER = insightface.model_zoo.get_model(
|
||||
model_path, providers=modules.globals.execution_providers
|
||||
|
|
|
|||
Loading…
Reference in New Issue