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

Deepfuze updated

Browse Source
This commit is contained in:
Sam Khoze
2024-07-10 20:10:17 +05:30
committed by GitHub
parent 13f89f47d8
commit e322a78210
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+5 -5
View File
@@ -12,7 +12,7 @@ import datetime
from typing import List from typing import List
import torch import torch
import psutil import psutil
import torchaudio
from PIL import Image, ExifTags from PIL import Image, ExifTags
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
from pathlib import Path from pathlib import Path
@@ -1018,7 +1018,7 @@ class DeepFuzeFaceSwap:
print(result.stderr) print(result.stderr)
audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav") audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav")
open(audio_file,"wb").write(audio()) torchaudio.save(audio_file,audio["waveform"],audio["sample_rate"])
subprocess.run(f"ffmpeg -i {faceswap_filename} -i {audio_file} -c copy {faceswap_filename.replace('.mp4','_.mp4')} -y".split()) subprocess.run(f"ffmpeg -i {faceswap_filename} -i {audio_file} -c copy {faceswap_filename.replace('.mp4','_.mp4')} -y".split())
return load_video_cv(faceswap_filename.replace('.mp4','_.mp4'),0,'Disabled',512,512,0,0,1) return load_video_cv(faceswap_filename.replace('.mp4','_.mp4'),0,'Disabled',512,512,0,0,1)
@@ -1336,7 +1336,7 @@ class DeepFuzeAdavance:
output_files.append(file_path) output_files.append(file_path)
audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav") audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav")
open(audio_file,"wb").write(audio()) torchaudio.save(audio_file,audio["waveform"],audio["sample_rate"])
print(audio_file) print(audio_file)
filename = os.path.join(result_dir,f"{str(time.time()).replace('.','')}.mp4") filename = os.path.join(result_dir,f"{str(time.time()).replace('.','')}.mp4")
enhanced_filename = os.path.join(result_dir,f"enhanced_{str(time.time()).replace('.','')}.mp4") enhanced_filename = os.path.join(result_dir,f"enhanced_{str(time.time()).replace('.','')}.mp4")
@@ -1513,7 +1513,7 @@ class TTS_generation:
language = supported_language.split("(")[1][:-1] language = supported_language.split("(")[1][:-1]
file_path = os.path.join(audio_path,str(time.time()).replace(".","")+".wav") file_path = os.path.join(audio_path,str(time.time()).replace(".","")+".wav")
open(file_path,"wb").write(audio()) torchaudio.save(audio_file,audio["waveform"],audio["sample_rate"])
command = [ command = [
'python', 'tts_generation.py', 'python', 'tts_generation.py',
'--model', checkpoint_path_voice, '--model', checkpoint_path_voice,
@@ -1531,7 +1531,7 @@ class TTS_generation:
print("stdout:", result.stdout) print("stdout:", result.stdout)
print("stderr:", result.stderr) print("stderr:", result.stderr)
audio = get_audio(file_path) audio = get_audio(file_path)
return (lambda : audio,) return (audio)