CalvinBackup/ComfyUI-DeepFuze
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Deepfuze windows support

Browse Source
This commit is contained in:
Sam Khoze
2024-06-16 23:32:26 +05:30
committed by GitHub
parent 77d1f557a8
commit 5f1d42751a
1 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+20 -8
View File
@@ -621,7 +621,7 @@ class DeepFuzeFaceSwap:
"face_mask_padding_right": ("INT",{"default":0,"min":0,"max":30,"step":1}),
"face_mask_padding_bottom": ("INT",{"default":0,"min":0,"max":30,"step":1}),
"face_mask_padding_top": ("INT",{"default":0,"min":0,"max":30,"step":1}),
"device" : (["cpu","gpu"],{"default":"cpu"}),
"device" : (["cpu","cuda","mps"],{"default":"cpu"}),
"frame_rate": (
"FLOAT",
{"default": 25, "min": 1, "step": 1},
@@ -950,7 +950,9 @@ class DeepFuzeFaceSwap:
]
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
print(command)
if platform == "win32":
@@ -975,7 +977,9 @@ class DeepFuzeFaceSwap:
faceswap_filename,
'--headless'
]
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
print(command)
if platform == "win32":
@@ -998,7 +1002,9 @@ class DeepFuzeFaceSwap:
'--headless'
]
print(command)
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
if platform == "win32":
result = subprocess.run(command,cwd="ComfyUI/custom_nodes/ComfyUI-DeepFuze",stdout=subprocess.PIPE)
@@ -1031,7 +1037,7 @@ class DeepFuzeAdavance:
"face_mask_padding_top": ("INT",{"default":0,"min":0,"max":30,"step":1}),
"trim_frame_start": ("INT",{"default":0,"max":2000},),
"trim_frame_end": ("INT",{"default":0,"max":2000},),
"device" : (["cpu","gpu"],{"default":"cpu"}),
"device" : (["cpu","cuda","mps"],{"default":"cpu"}),
"frame_rate": (
"FLOAT",
{"default": 25, "min": 1, "step": 1},
@@ -1350,7 +1356,9 @@ class DeepFuzeAdavance:
str(face_mask_padding_right),
'--headless'
]
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
print(command)
if platform == "win32":
@@ -1372,7 +1380,9 @@ class DeepFuzeAdavance:
enhanced_filename,
'--headless'
]
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
print(command)
if platform == "win32":
@@ -1396,7 +1406,9 @@ class DeepFuzeAdavance:
'--headless'
]
print(command)
if device=="gpu":
if device=="cuda":
command.extend(['--execution-providers',"cuda"])
elif device=="mps":
command.extend(['--execution-providers',"coreml"])
if platform == "win32":
result = subprocess.run(command,cwd="ComfyUI/custom_nodes/ComfyUI-DeepFuze",stdout=subprocess.PIPE)