Update faceswap.py
parent
8191992cb5
commit
b9071381a3
|
|
@ -20,7 +20,6 @@ import os
|
||||||
def get_models():
|
def get_models():
|
||||||
models_path = os.path.join(scripts.basedir(), "models/roop/*")
|
models_path = os.path.join(scripts.basedir(), "models/roop/*")
|
||||||
models = glob.glob(models_path)
|
models = glob.glob(models_path)
|
||||||
models += glob.glob(models_path)
|
|
||||||
models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")]
|
models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")]
|
||||||
return models
|
return models
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue