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