SamKhoze / ComfyUI-DeepFuze

DeepFuze is a state-of-the-art deep learning tool that seamlessly integrates with ComfyUI to revolutionize facial transformations, lipsyncing, Face Swapping, Lipsync Translation, video generation, and voice cloning.
https://cogidigm.com/
Apache License 2.0
320 stars 37 forks source link
comfyui-lipsync comfyui-nodes facefu faceswap lipsync sadta wav2l

πŸ‘‰πŸΌπŸ‘‰πŸΌπŸ‘‰πŸΌPlease take note of the following information: This repository is compatible and optimized for use with MAC CPU+MPS and Windows with CPU+CUDA. The installation process is not beginner-friendly for enabling CUDA Toolkit==11.8 and cuDNN (CUDA Deep Neural Network). If you are unsure about installing CUDA, you can still use the CPU, and no CUDA installation will be necessary. However, if you are able to install CUDA correctly, the processing speed will increase significantly.

DeepFuze

Watch the 4k Quality Video on YOUTUBE

DeepFuze Lipsync

Overview

DeepFuze is a state-of-the-art deep learning tool that seamlessly integrates with ComfyUI to revolutionize facial transformations, lipsyncing, video generation, voice cloning, face swapping, and lipsync translation. Leveraging advanced algorithms, DeepFuze enables users to combine audio and video with unparalleled realism, ensuring perfectly synchronized facial movements. This innovative solution is ideal for content creators, animators, developers, and anyone seeking to elevate their video editing projects with sophisticated AI-driven features.

Watch the 4k Quality Video on Youtube

DeepFuze Lipsync

Watch the 4k Quality Video on Youtube

IMAGE ALT TEXT HERE


Installation & Models Download


Windows Installation πŸ–₯️

Portable ComfyUI for Windows

Step 1

You must install Visual Studio, it works with the community version OR VS C++ Build Tools and select "Desktop Development with C++" under "Workloads -> Desktop & Mobile"

Step 2 (installing from ComfyUI-Manager)

From ComfyUI-Manager search for DeepFuze, and install the node. Restart your ComfyUI, and look at your terminal window to ensure there is no error, or Install from the ComfyUI manager, select "Install Via GIT URL", and copy past:

https://github.com/SamKhoze/CompfyUI-DeepFuze.git

GitInstall

Step 3

Restart your ComfyUI

IMPORTANT NOTE: CUDA INSTALLATION IS NOT BEGINNER-FRIENDLY, IF YOU DON'T KNOW WHAT YOU ARE DOING DO NOT TRY, USE THE CPU VERSION.

CUDA Installation

YOUTUBE LINK step by step instructions


Install Nvidia CUDA Toolkit==11.8 and cuDNN (CUDA Deep Neural Network) for Deep Learning, you must download the cuDNN version 8.9.2.26 from the NVIDIA DEVELOPER cuDNN Archive, if you don't have developer account you can directly download it from GoogleDrive. Ensure install Cuda1 1.8. I found this YOUTUBE video useful for installation. If you have a different version of CUDA here is a YOUTUBE link that guides you on how to uninstall your CUDA. Make sure to create paths in your Environment variable as described on YOUTUBE VIDEO Restart your computer after creating paths. Confirm your Cuda Installation, paste this code on your terminal window nvcc --version you should get a response like this:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0`

MAC Installation πŸ‘¨πŸ»β€πŸ’»

Do not install via ComfyUI-Manager it will not work, you must install it manually and follow the instructions below:

Video Tutorial How To Install on MAC YOUTUBE LINK

For MAC users please set the environment variable before running it

Activate your Virtual Environment, Conda or Venv

Install Pytorch

Here how to install and test your PyTorch

This method has been tested on a M1 and M3 Mac, You must run the below code on your terminal window for Mac Metal Performance Shaders (MPS) Apple's specialized solution for high-performance GPU programming on their devices. Integrating closely with the Metal framework, MPS provides a suite of highly optimized shaders for graphics and computing tasks, which is particularly beneficial in machine learning applications.

⚠️⚠️⚠️ Important Steps (if you miss these steps it will not work)

Copy and paste the command below to your terminal window.

export PYTORCH_ENABLE_MPS_FALLBACK=1

Mac users must INSTALL ONNX RUNTIME CPU instead of onnxruntime-gpu

pip install onnxruntime

macOS needs to install the original dlib.

pip install dlib 

Install Text to Speech for Voice Cloning Node

pip install TTS 

Navigate into custom_nodes folder

cd custom_nodes
git clone https://github.com/SamKhoze/CompfyUI-DeepFuze.git

Navigate into the CompfyUI-DeepFuze folder and install requirements.txt file

cd CompfyUI-DeepFuze
pip install -r requirements.txt

Prerequisites for Voice Cloning and Lipsyncing

Below are the two ComfyUI repositories required to load video and audio. Install them into your custom_nodes folder:

Clone the repositories:

cd custom_nodes
git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git

Errors 😾

CV Error: If you encounter the error "ComfyUI_windows_portable\ComfyUI\output\deepfuze\faceswap_file.mp4 could not be loaded with cv," it means that onnxruntime is not installed. To fix this, make sure to install onnxruntime for CPU and onnxruntime-gpu for Windows. Mac users should upgrade OpenCV using the command pip install --upgrade opencv-python-headless in their virtual environment. For Windows users, go to ComfyUI Manager, click on "pip install," paste --upgrade opencv-python-headless, click OK, and restart your ComfyUI.


Missing zlibwapi.dll error: Search for NVIDIA zlibwapi.dll file, download it and copy it in C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin\zlibwapi.dll


If you get an error installing TTS, it is most likely because you have different versions of Python, make sure to install the correct version


If you get an error: ImportError: cannot import name 'get_full_repo_name' from 'huggingface_hub' Run the below codes on your terminal it will solve the issue

conda install chardet 
pip install --upgrade transformers==4.39.2 

if you get any error for any packages, open the requirements.txt file with any text editor remove the version from the front of the package name, and reinstall requirments.txt again

Models🌟🌟🌟

You can download models directly from GoogleDrive and place models into the PATH ./ComfyUI/models/deepfuze/ Ensure to manually download each model one by one and place them, due to the size of the models some of the models won't download if you download the folder preparing the environmental variables navigate into your custom_nodes folder and git clone or manually download the code and extract it into the custom_nodes folder.

OpenAI API setup for voice cloning (Optional)


You need an OpenAI API Key if you wish to use the "DeepFuze Openai LLM" node for generating dialogues for voice cloning

To use the "Openai LLM" node for voice cloning dialogues, you need an OpenAI API Key. You can get this key and set it up by following the instructions in the OpenAI Developer quickstart guide. Please note that the "Openai LLM" node does not save your API key. Every time you close the node, you will need to manually copy and paste your API key. You can also add the API key as an Environment Variable using the following commands: For Windows: setx OPENAI_API_KEY "your-api-key-here", and for Mac: export OPENAI_API_KEY='your-api-key-here'. The next time you need to copy and paste your API key into the LLM Node, you can type the following command in your terminal: echo $OPENAI_API_KEY, and it will print your API Key, allowing you to copy and paste it into your Openai LLM node.

Nodes

DeepFuze Nodes Overview YOUTUBE LINK

DeepFuze Lipsync Node 🫦

DeepFuze Lipsync DeepFuze Lipsync Node example

This node generates lipsyncing video from, video, image, and audio files. For higher quality export the IMAGE output as an image batch instead of a video combined, you can get up to 4k quality image size. IMPORTANT: You must load audio with the "VHS load audio" node from the VideoHelperSuit node.

Input Types:

Output Types:

DeepFuze Lipsync Features:


DeepFuze FaceSwap Node 🎭

DeepFuze FaceSwap Node DeepFuze FaceSwap Node example

DeepFuze FaceSwap example

This node Swaps, Enhances, and Restores faces from, video, and image. or higher quality export the IMAGE output as an image batch instead of a video combined, you can get up to 4k quality image size.

Input Types:

Output Types:

DeepFuze FaceSwap Features:

Face Detector Model Summary Table (RetinaFace provides higher quality by incorporating contextual information around the face, which helps in detecting faces under various conditions, such as occlusions, different scales, and poses.

Feature YOLOFace RetinaFace SCRFD YuNet
Architecture Single-shot YOLO Single-stage RetinaNet Single-stage Cascade Lightweight Custom
Speed Very Fast Moderate Fast Very Fast
Accuracy Good Very High High Good
Robustness Moderate Very High High Moderate
Computational Efficiency High Moderate High Very High
Use Cases Real-time, less complex scenes High-precision, robust needs Balanced, mobile/edge devices Mobile, embedded, real-time
Pros Speed Accuracy, robustness Efficiency, accuracy Lightweight, efficient
Cons Accuracy trade-offs Computationally heavy Not the fastest Less robust in complex scenes

DeepFuze_TTS Node (Voice Cloning) πŸŽ™οΈ

DeepFuze TTS_Node

Languages:

DeepFuze_TTS voice cloning supports 17 languages: English (en), Spanish (es), French (fr), German (de), Italian (it), Portuguese (pt), Polish (pl), Turkish (tr), Russian (ru), Dutch (nl), Czech (cs), Arabic (ar), Chinese (zh-cn), Japanese (ja), Hungarian (hu), Korean (ko) Hindi (hi).

This node is used to clone any voice from typed input. The audio file should be 10-15 seconds long for better results and should not have much noise. To avoid any sample rate error, load MP3 audio and only work with AudioScheduler node. We are working on developing a converter node to solve this issue.

Input Types:

Output Types:


DeepFuze Openai LLM Node πŸ€–

DeepFuze Openai_Node DeepFuze Openai_Node

The "LLM Integration" node is used to incorporate LLM (Language Model) into the voice cloning process. You can input your dialogue and configure parameters, and the AI-generated texts will be employed for voice cloning. Furthermore, you can utilize this node in place of ChatGPT to produce text from LLM or to ask any questions in the same manner as you would with ChatGPT. You can view the output of the DeepFuze_LLM by connecting the LLM_RESPONSE to the "Display Any" node from rgthree-comfy this node also can be used for prompt generations and any nodes input texts.

Input Types:

Output Types:

DeepFuze Openai LLM Features:


DeepFuze Padding Node πŸ‘Ί

DeepFuze Padding_Node

Input Types:

DeepFuze Padding Features:


DeepFuze Save Audio (Playback) Node πŸ”‰

DeepFuze Save_Audio_Node DeepFuze Save_Audio This node is used to save the output of the "Voice Cloning" node. Additionally, you can trim the audio and play it back.

Input Types:

DeepFuze Padding Features:


Basic Integrations

Voice Cloning + Lipsync Generation

BasicWorkspace

Voice Cloning + Lipsync Generation + FaceSwap

BasicWorkspace


Repository Structure

ComfyUI-DeepFuze/
β”œβ”€β”€ __init__.py
β”œβ”€β”€ __pycache__/
β”‚   β”œβ”€β”€ __init__.cpython-311.pyc
β”‚   β”œβ”€β”€ audio_playback.cpython-311.pyc
β”‚   β”œβ”€β”€ llm_node.cpython-311.pyc
β”‚   β”œβ”€β”€ nodes.cpython-311.pyc
β”‚   └── utils.cpython-311.pyc
β”œβ”€β”€ audio_playback.py
β”œβ”€β”€ deepfuze/
β”‚   β”œβ”€β”€ __init__.py
β”‚   β”œβ”€β”€ audio.py
β”‚   β”œβ”€β”€ choices.py
β”‚   β”œβ”€β”€ common_helper.py
β”‚   β”œβ”€β”€ config.py
β”‚   β”œβ”€β”€ content_analyser.py
β”‚   β”œβ”€β”€ core.py
β”‚   β”œβ”€β”€ download.py
β”‚   β”œβ”€β”€ execution.py
β”‚   β”œβ”€β”€ face_analyser.py
β”‚   β”œβ”€β”€ face_helper.py
β”‚   β”œβ”€β”€ face_masker.py
β”‚   β”œβ”€β”€ face_store.py
β”‚   β”œβ”€β”€ ffmpeg.py
β”‚   β”œβ”€β”€ filesystem.py
β”‚   β”œβ”€β”€ globals.py
β”‚   β”œβ”€β”€ installer.py
β”‚   β”œβ”€β”€ logger.py
β”‚   β”œβ”€β”€ memory.py
β”‚   β”œβ”€β”€ metadata.py
β”‚   β”œβ”€β”€ normalizer.py
β”‚   β”œβ”€β”€ process_manager.py
β”œβ”€β”€ requirements.txt
β”œβ”€β”€ images/
β”œβ”€β”€ install.py
β”œβ”€β”€ LICENSE.txt
β”œβ”€β”€ llm_node.py
β”œβ”€β”€ mypy.ini
β”œβ”€β”€ nodes.py
β”œβ”€β”€ README.md
β”œβ”€β”€ requirements.txt
β”œβ”€β”€ run.py
β”œβ”€β”€ tests/
β”‚   β”œβ”€β”€ __init__.py
β”‚   β”œβ”€β”€ test_audio.py
β”‚   β”œβ”€β”€ test_cli_face_debugger.py
β”‚   β”œβ”€β”€ test_cli_face_enhancer.py
β”‚   β”œβ”€β”€ test_cli_face_swapper.py
β”‚   β”œβ”€β”€ test_cli_frame_colorizer.py
β”‚   β”œβ”€β”€ test_cli_frame_enhancer.py
β”‚   β”œβ”€β”€ test_cli_lip_syncer.py
β”‚   β”œβ”€β”€ test_common_helper.py
β”‚   β”œβ”€β”€ test_config.py
β”‚   β”œβ”€β”€ test_download.py
β”‚   β”œβ”€β”€ test_execution.py
β”‚   β”œβ”€β”€ test_face_analyser.py
β”‚   β”œβ”€β”€ test_ffmpeg.py
β”‚   β”œβ”€β”€ test_filesystem.py
β”‚   β”œβ”€β”€ test_memory.py
β”‚   β”œβ”€β”€ test_normalizer.py
β”‚   β”œβ”€β”€ test_process_manager.py
β”‚   β”œβ”€β”€ test_vision.py
β”‚   └── test_wording.py
β”œβ”€β”€ tts_generation.py
└── utils.py

Example of How to Use DeepFuze Programmatically

from deepfuze import DeepFuze

# Initialize the DeepFuze instance
deepfuze = DeepFuze()

# Load video and audio files
deepfuze.load_video('path/to/video.mp4')
deepfuze.load_audio('path/to/audio.mp3')
deepfuze.load_checkpoint('path/to/checkpoint_path')

# Set parameters (optional)
deepfuze.set_parameters(sync_level=5, transform_intensity=3)

# Generate lipsynced video
output_path = deepfuze.generate(output='path/to/output.mp4')

print(f"Lipsynced video saved at {output_path}")

Acknowledgements

This repository could not have been completed without the contributions from FaceFusion, InsightFace,SadTalker, Facexlib, GFPGAN, GPEN, Real-ESRGAN, TTS, SSD, and wav2lip,

  1. Please carefully read and comply with the open-source license applicable to this code and models before using it.
  2. Please carefully read and comply with the intellectual property declaration applicable to this code and models before using it.
  3. This open-source code runs completely offline and does not collect any personal information or other data. If you use this code to provide services to end-users and collect related data, please take necessary compliance measures according to applicable laws and regulations (such as publishing privacy policies, adopting necessary data security strategies, etc.). If the collected data involves personal information, user consent must be obtained (if applicable).
  4. It is prohibited to use this open-source code for activities that harm the legitimate rights and interests of others (including but not limited to fraud, deception, infringement of others' portrait rights, reputation rights, etc.), or other behaviors that violate applicable laws and regulations or go against social ethics and good customs (including providing incorrect or false information, terrorist, child/minors pornography and violent information, etc.). Otherwise, you may be liable for legal responsibilities.

The DeepFuze code is developed by Dr. Sam Khoze and his team. Feel free to use the DeepFuze code for personal, research, academic, and commercial purposes. You can create videos with this tool, but please make sure to follow local laws and use it responsibly. The developers will not be responsible for any misuse of the tool by users.