Compare commits
5 Commits
22bb14e9ad
...
36095d5493
Author | SHA1 | Date |
---|---|---|
|
36095d5493 | |
|
2b70131e6a | |
|
647c5f250f | |
|
ae88412aae | |
|
b7e011f5e7 |
|
@ -28,11 +28,21 @@ models_dir = os.path.join(
|
||||||
|
|
||||||
|
|
||||||
def pre_check() -> bool:
|
def pre_check() -> bool:
|
||||||
download_directory_path = abs_dir
|
# Use models_dir instead of abs_dir to save to the correct location
|
||||||
|
download_directory_path = models_dir
|
||||||
|
|
||||||
|
# Make sure the models directory exists, catch permission errors if they occur
|
||||||
|
try:
|
||||||
|
os.makedirs(download_directory_path, exist_ok=True)
|
||||||
|
except OSError as e:
|
||||||
|
logging.error(f"Failed to create directory {download_directory_path} due to permission error: {e}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Use the direct download URL from Hugging Face
|
||||||
conditional_download(
|
conditional_download(
|
||||||
download_directory_path,
|
download_directory_path,
|
||||||
[
|
[
|
||||||
"https://huggingface.co/hacksider/deep-live-cam/blob/main/inswapper_128_fp16.onnx"
|
"https://huggingface.co/hacksider/deep-live-cam/resolve/main/inswapper_128_fp16.onnx"
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
return True
|
return True
|
||||||
|
|
|
@ -14,7 +14,7 @@ torch; sys_platform != 'darwin'
|
||||||
torch==2.5.1; sys_platform == 'darwin'
|
torch==2.5.1; sys_platform == 'darwin'
|
||||||
torchvision; sys_platform != 'darwin'
|
torchvision; sys_platform != 'darwin'
|
||||||
torchvision==0.20.1; sys_platform == 'darwin'
|
torchvision==0.20.1; sys_platform == 'darwin'
|
||||||
onnxruntime-silicon==1.21.0; sys_platform == 'darwin' and platform_machine == 'arm64'
|
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
|
||||||
onnxruntime-gpu==1.22.0; sys_platform != 'darwin'
|
onnxruntime-gpu==1.22.0; sys_platform != 'darwin'
|
||||||
tensorflow; sys_platform != 'darwin'
|
tensorflow; sys_platform != 'darwin'
|
||||||
opennsfw2==0.10.2
|
opennsfw2==0.10.2
|
||||||
|
|
Loading…
Reference in New Issue