├── Hallo_node.py ├── LICENSE ├── README.md ├── __init__.py ├── accelerate_config.yaml ├── assets ├── framework_1.jpg ├── framework_2.jpg └── wechat.jpeg ├── configs ├── inference │ └── long.yaml ├── train │ ├── stage1.yaml │ ├── stage2_long.yaml │ └── video_sr.yaml ├── unet │ ├── config.json │ └── unet.yaml └── vae │ └── config.json ├── example.png ├── hallo ├── __init__.py ├── animate │ ├── __init__.py │ ├── face_animate.py │ └── face_animate_static.py ├── basicsr │ ├── VERSION │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── data_sampler.py │ │ ├── data_util.py │ │ ├── gaussian_kernels.py │ │ ├── prefetch_dataloader.py │ │ ├── transforms.py │ │ └── vfhq_dataset.py │ ├── hallo_archs │ │ ├── __init__.py │ │ ├── arcface_arch.py │ │ ├── arch_util.py │ │ ├── codeformer_arch.py │ │ ├── rrdbnet_arch.py │ │ ├── vgg_arch.py │ │ └── vqgan_arch.py │ ├── losses │ │ ├── __init__.py │ │ ├── loss_util.py │ │ └── losses.py │ ├── metrics │ │ ├── __init__.py │ │ ├── metric_util.py │ │ └── psnr_ssim.py │ ├── models │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── codeformer_temporal_model.py │ │ ├── lr_scheduler.py │ │ ├── sr_model.py │ │ └── vqgan_model.py │ ├── ops │ │ ├── __init__.py │ │ ├── dcn │ │ │ ├── __init__.py │ │ │ ├── deform_conv.py │ │ │ └── src │ │ │ │ ├── deform_conv_cuda.cpp │ │ │ │ ├── deform_conv_cuda_kernel.cu │ │ │ │ └── deform_conv_ext.cpp │ │ ├── fused_act │ │ │ ├── __init__.py │ │ │ ├── fused_act.py │ │ │ └── src │ │ │ │ ├── fused_bias_act.cpp │ │ │ │ └── fused_bias_act_kernel.cu │ │ └── upfirdn2d │ │ │ ├── __init__.py │ │ │ ├── src │ │ │ ├── upfirdn2d.cpp │ │ │ └── upfirdn2d_kernel.cu │ │ │ └── upfirdn2d.py │ ├── setup.py │ ├── train.py │ ├── utils │ │ ├── __init__.py │ │ ├── dist_util.py │ │ ├── download_util.py │ │ ├── file_client.py │ │ ├── img_util.py │ │ ├── lmdb_util.py │ │ ├── logger.py │ │ ├── matlab_functions.py │ │ ├── misc.py │ │ ├── options.py │ │ ├── realesrgan_utils.py │ │ ├── registry.py │ │ └── video_util.py │ └── version.py ├── datasets │ ├── __init__.py │ ├── audio_processor.py │ ├── image_processor.py │ ├── mask_image.py │ └── talk_video.py ├── facelib │ ├── detection │ │ ├── __init__.py │ │ ├── align_trans.py │ │ ├── matlab_cp2tform.py │ │ ├── retinaface │ │ │ ├── retinaface.py │ │ │ ├── retinaface_net.py │ │ │ └── retinaface_utils.py │ │ └── yolov5face │ │ │ ├── __init__.py │ │ │ ├── face_detector.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── experimental.py │ │ │ ├── yolo.py │ │ │ ├── yolov5l.yaml │ │ │ └── yolov5n.yaml │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── autoanchor.py │ │ │ ├── datasets.py │ │ │ ├── extract_ckpt.py │ │ │ ├── general.py │ │ │ └── torch_utils.py │ ├── parsing │ │ ├── __init__.py │ │ ├── bisenet.py │ │ ├── parsenet.py │ │ └── resnet.py │ └── utils │ │ ├── __init__.py │ │ ├── face_restoration_helper.py │ │ ├── face_utils.py │ │ └── misc.py ├── models │ ├── __init__.py │ ├── attention.py │ ├── audio_proj.py │ ├── face_locator.py │ ├── image_proj.py │ ├── motion_module.py │ ├── mutual_self_attention.py │ ├── resnet.py │ ├── transformer_2d.py │ ├── transformer_3d.py │ ├── unet_2d_blocks.py │ ├── unet_2d_condition.py │ ├── unet_3d.py │ ├── unet_3d_blocks.py │ └── wav2vec.py ├── utils │ ├── __init__.py │ ├── config.py │ └── util.py └── video_sr.py ├── pyproject.toml ├── requirements.txt ├── scripts ├── app.py ├── data_preprocess.py ├── extract_meta_info_stage1.py ├── extract_meta_info_stage2.py ├── inference_long.py ├── train_stage1.py └── train_stage2_long.py ├── utils.py └── workflow.json /Hallo_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/Hallo_node.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/__init__.py -------------------------------------------------------------------------------- /accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/accelerate_config.yaml -------------------------------------------------------------------------------- /assets/framework_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/assets/framework_1.jpg -------------------------------------------------------------------------------- /assets/framework_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/assets/framework_2.jpg -------------------------------------------------------------------------------- /assets/wechat.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/assets/wechat.jpeg -------------------------------------------------------------------------------- /configs/inference/long.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/inference/long.yaml -------------------------------------------------------------------------------- /configs/train/stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/train/stage1.yaml -------------------------------------------------------------------------------- /configs/train/stage2_long.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/train/stage2_long.yaml -------------------------------------------------------------------------------- /configs/train/video_sr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/train/video_sr.yaml -------------------------------------------------------------------------------- /configs/unet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/unet/config.json -------------------------------------------------------------------------------- /configs/unet/unet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/unet/unet.yaml -------------------------------------------------------------------------------- /configs/vae/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/configs/vae/config.json -------------------------------------------------------------------------------- /example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/example.png -------------------------------------------------------------------------------- /hallo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/animate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/animate/face_animate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/animate/face_animate.py -------------------------------------------------------------------------------- /hallo/animate/face_animate_static.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/animate/face_animate_static.py -------------------------------------------------------------------------------- /hallo/basicsr/VERSION: -------------------------------------------------------------------------------- 1 | 1.3.2 2 | -------------------------------------------------------------------------------- /hallo/basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /hallo/basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/data_util.py -------------------------------------------------------------------------------- /hallo/basicsr/data/gaussian_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/gaussian_kernels.py -------------------------------------------------------------------------------- /hallo/basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /hallo/basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/transforms.py -------------------------------------------------------------------------------- /hallo/basicsr/data/vfhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/data/vfhq_dataset.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/arcface_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/arcface_arch.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/arch_util.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/codeformer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/codeformer_arch.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/rrdbnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/rrdbnet_arch.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/vgg_arch.py -------------------------------------------------------------------------------- /hallo/basicsr/hallo_archs/vqgan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/hallo_archs/vqgan_arch.py -------------------------------------------------------------------------------- /hallo/basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /hallo/basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/losses/losses.py -------------------------------------------------------------------------------- /hallo/basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /hallo/basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /hallo/basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/base_model.py -------------------------------------------------------------------------------- /hallo/basicsr/models/codeformer_temporal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/codeformer_temporal_model.py -------------------------------------------------------------------------------- /hallo/basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /hallo/basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /hallo/basicsr/models/vqgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/models/vqgan_model.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /hallo/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /hallo/basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /hallo/basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /hallo/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /hallo/basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /hallo/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /hallo/basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /hallo/basicsr/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/setup.py -------------------------------------------------------------------------------- /hallo/basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/train.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/logger.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/misc.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/options.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/realesrgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/realesrgan_utils.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/registry.py -------------------------------------------------------------------------------- /hallo/basicsr/utils/video_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/utils/video_util.py -------------------------------------------------------------------------------- /hallo/basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/basicsr/version.py -------------------------------------------------------------------------------- /hallo/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/datasets/audio_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/datasets/audio_processor.py -------------------------------------------------------------------------------- /hallo/datasets/image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/datasets/image_processor.py -------------------------------------------------------------------------------- /hallo/datasets/mask_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/datasets/mask_image.py -------------------------------------------------------------------------------- /hallo/datasets/talk_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/datasets/talk_video.py -------------------------------------------------------------------------------- /hallo/facelib/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/__init__.py -------------------------------------------------------------------------------- /hallo/facelib/detection/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/align_trans.py -------------------------------------------------------------------------------- /hallo/facelib/detection/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/matlab_cp2tform.py -------------------------------------------------------------------------------- /hallo/facelib/detection/retinaface/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/retinaface/retinaface.py -------------------------------------------------------------------------------- /hallo/facelib/detection/retinaface/retinaface_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/retinaface/retinaface_net.py -------------------------------------------------------------------------------- /hallo/facelib/detection/retinaface/retinaface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/retinaface/retinaface_utils.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/face_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/face_detector.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/models/common.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/models/experimental.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/models/yolo.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/models/yolov5l.yaml -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/models/yolov5n.yaml -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/utils/autoanchor.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/utils/datasets.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/extract_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/utils/extract_ckpt.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/utils/general.py -------------------------------------------------------------------------------- /hallo/facelib/detection/yolov5face/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/detection/yolov5face/utils/torch_utils.py -------------------------------------------------------------------------------- /hallo/facelib/parsing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/parsing/__init__.py -------------------------------------------------------------------------------- /hallo/facelib/parsing/bisenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/parsing/bisenet.py -------------------------------------------------------------------------------- /hallo/facelib/parsing/parsenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/parsing/parsenet.py -------------------------------------------------------------------------------- /hallo/facelib/parsing/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/parsing/resnet.py -------------------------------------------------------------------------------- /hallo/facelib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/utils/__init__.py -------------------------------------------------------------------------------- /hallo/facelib/utils/face_restoration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/utils/face_restoration_helper.py -------------------------------------------------------------------------------- /hallo/facelib/utils/face_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/utils/face_utils.py -------------------------------------------------------------------------------- /hallo/facelib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/facelib/utils/misc.py -------------------------------------------------------------------------------- /hallo/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/attention.py -------------------------------------------------------------------------------- /hallo/models/audio_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/audio_proj.py -------------------------------------------------------------------------------- /hallo/models/face_locator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/face_locator.py -------------------------------------------------------------------------------- /hallo/models/image_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/image_proj.py -------------------------------------------------------------------------------- /hallo/models/motion_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/motion_module.py -------------------------------------------------------------------------------- /hallo/models/mutual_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/mutual_self_attention.py -------------------------------------------------------------------------------- /hallo/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/resnet.py -------------------------------------------------------------------------------- /hallo/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/transformer_2d.py -------------------------------------------------------------------------------- /hallo/models/transformer_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/transformer_3d.py -------------------------------------------------------------------------------- /hallo/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /hallo/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/unet_2d_condition.py -------------------------------------------------------------------------------- /hallo/models/unet_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/unet_3d.py -------------------------------------------------------------------------------- /hallo/models/unet_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/unet_3d_blocks.py -------------------------------------------------------------------------------- /hallo/models/wav2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/models/wav2vec.py -------------------------------------------------------------------------------- /hallo/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hallo/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/utils/config.py -------------------------------------------------------------------------------- /hallo/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/utils/util.py -------------------------------------------------------------------------------- /hallo/video_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/hallo/video_sr.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/app.py -------------------------------------------------------------------------------- /scripts/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/data_preprocess.py -------------------------------------------------------------------------------- /scripts/extract_meta_info_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/extract_meta_info_stage1.py -------------------------------------------------------------------------------- /scripts/extract_meta_info_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/extract_meta_info_stage2.py -------------------------------------------------------------------------------- /scripts/inference_long.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/inference_long.py -------------------------------------------------------------------------------- /scripts/train_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/train_stage1.py -------------------------------------------------------------------------------- /scripts/train_stage2_long.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/scripts/train_stage2_long.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/utils.py -------------------------------------------------------------------------------- /workflow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_Hallo2/HEAD/workflow.json --------------------------------------------------------------------------------