├── LICENSE ├── README.md ├── __init__.py ├── diffueraser_node.py ├── example_workflows ├── differaser.json ├── example.png └── example1.png ├── examples ├── mask.mp4 └── video.mp4 ├── libs ├── __init__.py ├── brushnet_CA.py ├── diffueraser.py ├── pipeline_diffueraser.py ├── transformer_temporal.py ├── unet_2d_blocks.py ├── unet_2d_condition.py ├── unet_3d_blocks.py ├── unet_motion_model.py └── v1-inference.yaml ├── node_utils.py ├── propainter ├── RAFT │ ├── __init__.py │ ├── corr.py │ ├── datasets.py │ ├── demo.py │ ├── extractor.py │ ├── raft.py │ ├── update.py │ └── utils │ │ ├── __init__.py │ │ ├── augmentor.py │ │ ├── flow_viz.py │ │ ├── flow_viz_pt.py │ │ ├── frame_utils.py │ │ └── utils.py ├── core │ ├── __init__.py │ ├── dataset.py │ ├── dist.py │ ├── loss.py │ ├── lr_scheduler.py │ ├── metrics.py │ ├── prefetch_dataloader.py │ ├── trainer.py │ ├── trainer_flow_w_edge.py │ └── utils.py ├── inference.py ├── model │ ├── __init__.py │ ├── canny │ │ ├── __init__.py │ │ ├── canny_filter.py │ │ ├── filter.py │ │ ├── gaussian.py │ │ ├── kernels.py │ │ └── sobel.py │ ├── misc.py │ ├── modules │ │ ├── __init__.py │ │ ├── base_module.py │ │ ├── deformconv.py │ │ ├── flow_comp_raft.py │ │ ├── flow_loss_utils.py │ │ ├── sparse_transformer.py │ │ └── spectral_norm.py │ ├── propainter.py │ ├── recurrent_flow_completion.py │ └── vgg_arch.py └── utils │ ├── __init__.py │ ├── download_util.py │ ├── file_client.py │ ├── flow_util.py │ └── img_util.py ├── pyproject.toml ├── requirements.txt ├── run_diffueraser.py └── sd15_repo ├── feature_extractor └── preprocessor_config.json ├── model_index.json ├── safety_checker └── config.json ├── scheduler └── scheduler_config.json ├── text_encoder └── config.json ├── tokenizer ├── merges.txt ├── special_tokens_map.json ├── tokenizer_config.json └── vocab.json ├── unet └── config.json └── vae └── config.json /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | 2 | from .diffueraser_node import * -------------------------------------------------------------------------------- /diffueraser_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/diffueraser_node.py -------------------------------------------------------------------------------- /example_workflows/differaser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/example_workflows/differaser.json -------------------------------------------------------------------------------- /example_workflows/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/example_workflows/example.png -------------------------------------------------------------------------------- /example_workflows/example1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/example_workflows/example1.png -------------------------------------------------------------------------------- /examples/mask.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/examples/mask.mp4 -------------------------------------------------------------------------------- /examples/video.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/examples/video.mp4 -------------------------------------------------------------------------------- /libs/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /libs/brushnet_CA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/brushnet_CA.py -------------------------------------------------------------------------------- /libs/diffueraser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/diffueraser.py -------------------------------------------------------------------------------- /libs/pipeline_diffueraser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/pipeline_diffueraser.py -------------------------------------------------------------------------------- /libs/transformer_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/transformer_temporal.py -------------------------------------------------------------------------------- /libs/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/unet_2d_blocks.py -------------------------------------------------------------------------------- /libs/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/unet_2d_condition.py -------------------------------------------------------------------------------- /libs/unet_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/unet_3d_blocks.py -------------------------------------------------------------------------------- /libs/unet_motion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/unet_motion_model.py -------------------------------------------------------------------------------- /libs/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/libs/v1-inference.yaml -------------------------------------------------------------------------------- /node_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/node_utils.py -------------------------------------------------------------------------------- /propainter/RAFT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/__init__.py -------------------------------------------------------------------------------- /propainter/RAFT/corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/corr.py -------------------------------------------------------------------------------- /propainter/RAFT/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/datasets.py -------------------------------------------------------------------------------- /propainter/RAFT/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/demo.py -------------------------------------------------------------------------------- /propainter/RAFT/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/extractor.py -------------------------------------------------------------------------------- /propainter/RAFT/raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/raft.py -------------------------------------------------------------------------------- /propainter/RAFT/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/update.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/__init__.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/augmentor.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/flow_viz.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/flow_viz_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/flow_viz_pt.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/frame_utils.py -------------------------------------------------------------------------------- /propainter/RAFT/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/RAFT/utils/utils.py -------------------------------------------------------------------------------- /propainter/core/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /propainter/core/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/dataset.py -------------------------------------------------------------------------------- /propainter/core/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/dist.py -------------------------------------------------------------------------------- /propainter/core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/loss.py -------------------------------------------------------------------------------- /propainter/core/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/lr_scheduler.py -------------------------------------------------------------------------------- /propainter/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/metrics.py -------------------------------------------------------------------------------- /propainter/core/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/prefetch_dataloader.py -------------------------------------------------------------------------------- /propainter/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/trainer.py -------------------------------------------------------------------------------- /propainter/core/trainer_flow_w_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/trainer_flow_w_edge.py -------------------------------------------------------------------------------- /propainter/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/core/utils.py -------------------------------------------------------------------------------- /propainter/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/inference.py -------------------------------------------------------------------------------- /propainter/model/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /propainter/model/canny/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /propainter/model/canny/canny_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/canny/canny_filter.py -------------------------------------------------------------------------------- /propainter/model/canny/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/canny/filter.py -------------------------------------------------------------------------------- /propainter/model/canny/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/canny/gaussian.py -------------------------------------------------------------------------------- /propainter/model/canny/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/canny/kernels.py -------------------------------------------------------------------------------- /propainter/model/canny/sobel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/canny/sobel.py -------------------------------------------------------------------------------- /propainter/model/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/misc.py -------------------------------------------------------------------------------- /propainter/model/modules/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /propainter/model/modules/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/base_module.py -------------------------------------------------------------------------------- /propainter/model/modules/deformconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/deformconv.py -------------------------------------------------------------------------------- /propainter/model/modules/flow_comp_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/flow_comp_raft.py -------------------------------------------------------------------------------- /propainter/model/modules/flow_loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/flow_loss_utils.py -------------------------------------------------------------------------------- /propainter/model/modules/sparse_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/sparse_transformer.py -------------------------------------------------------------------------------- /propainter/model/modules/spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/modules/spectral_norm.py -------------------------------------------------------------------------------- /propainter/model/propainter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/propainter.py -------------------------------------------------------------------------------- /propainter/model/recurrent_flow_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/recurrent_flow_completion.py -------------------------------------------------------------------------------- /propainter/model/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/model/vgg_arch.py -------------------------------------------------------------------------------- /propainter/utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /propainter/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/utils/download_util.py -------------------------------------------------------------------------------- /propainter/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/utils/file_client.py -------------------------------------------------------------------------------- /propainter/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/utils/flow_util.py -------------------------------------------------------------------------------- /propainter/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/propainter/utils/img_util.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_diffueraser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/run_diffueraser.py -------------------------------------------------------------------------------- /sd15_repo/feature_extractor/preprocessor_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/feature_extractor/preprocessor_config.json -------------------------------------------------------------------------------- /sd15_repo/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/model_index.json -------------------------------------------------------------------------------- /sd15_repo/safety_checker/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/safety_checker/config.json -------------------------------------------------------------------------------- /sd15_repo/scheduler/scheduler_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/scheduler/scheduler_config.json -------------------------------------------------------------------------------- /sd15_repo/text_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/text_encoder/config.json -------------------------------------------------------------------------------- /sd15_repo/tokenizer/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/tokenizer/merges.txt -------------------------------------------------------------------------------- /sd15_repo/tokenizer/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/tokenizer/special_tokens_map.json -------------------------------------------------------------------------------- /sd15_repo/tokenizer/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/tokenizer/tokenizer_config.json -------------------------------------------------------------------------------- /sd15_repo/tokenizer/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/tokenizer/vocab.json -------------------------------------------------------------------------------- /sd15_repo/unet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/unet/config.json -------------------------------------------------------------------------------- /sd15_repo/vae/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_DiffuEraser/HEAD/sd15_repo/vae/config.json --------------------------------------------------------------------------------