CalvinBackup/sd-webui-faceswaplab
1
0
Fork 0
Code Issues 56 Pull Requests 4 Actions Packages Projects Releases 8 Wiki Activity

add gender selection in build

Browse Source
This commit is contained in:
Tran Xen
2023-09-09 16:25:06 +02:00
parent c94d2aab4d
commit bba4845dbf
6 changed files with 57 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/faceswaplab_utils/models_utils.py
+3 -3
View File
@@ -73,14 +73,14 @@ def get_current_swap_model() -> str:
model = models[0] if len(models) else None
logger.info("Try to use model : %s", model)
try:
if not os.path.isfile(model): # type: ignore
if not model or not os.path.isfile(model): # type: ignore
logger.error("The model %s cannot be found or loaded", model)
raise FileNotFoundError(
"No faceswap model found. Please add it to the faceswaplab directory."
"No faceswap model found. Please add it to the faceswaplab directory. Ensure the model is in the proper directory (<sdwebui>/models/faceswaplab/inswapper_128.onnx)"
)
except:
raise FileNotFoundError(
"Was not able to check model, please ensure the model is in the proper directory"
"No faceswap model found. Please add it to the faceswaplab directory. Ensure the model is in the proper directory (<sdwebui>/models/faceswaplab/inswapper_128.onnx)"
)
assert model is not None