├── .gitignore ├── LICENSE ├── README.md ├── TRAIN_AND_VALIDATE.md ├── assets ├── gt_5544.jpg ├── logo.jpg └── report.pdf ├── causalvideovae ├── __init__.py ├── dataset │ ├── __init__.py │ ├── ddp_sampler.py │ ├── transform.py │ └── video_dataset.py ├── eval │ ├── __init__.py │ ├── cal_fvd.py │ ├── cal_lpips.py │ ├── cal_psnr.py │ ├── cal_ssim.py │ └── fvd │ │ ├── styleganv │ │ └── fvd.py │ │ └── videogpt │ │ ├── fvd.py │ │ └── pytorch_i3d.py ├── model │ ├── __init__.py │ ├── configuration_videobase.py │ ├── ema_model.py │ ├── losses │ │ ├── __init__.py │ │ ├── discriminator.py │ │ ├── lpips.py │ │ └── perceptual_loss.py │ ├── modeling_output.py │ ├── modeling_videobase.py │ ├── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── conv.py │ │ ├── normalize.py │ │ ├── ops.py │ │ ├── quant.py │ │ ├── resnet_block.py │ │ ├── updownsample.py │ │ └── wavelet.py │ ├── registry.py │ ├── utils │ │ ├── __init__.py │ │ ├── distrib_utils.py │ │ ├── module_utils.py │ │ ├── scheduler_utils.py │ │ └── wavelet_utils.py │ └── vae │ │ ├── __init__.py │ │ ├── modeling_causalvae.py │ │ ├── modeling_wfvae.py │ │ └── modeling_wfvae2.py └── utils │ ├── __init__.py │ ├── downloader.py │ ├── taming_download.py │ └── video_utils.py ├── examples ├── accelerate_configs │ ├── ddp_config.yaml │ ├── debug.yaml │ ├── deepspeed_zero2_config.yaml │ ├── deepspeed_zero2_offload_config.yaml │ ├── deepspeed_zero3_config.yaml │ ├── deepspeed_zero3_offload_config.yaml │ ├── default_config.yaml │ ├── hostfile │ ├── hostfile1 │ ├── hostfile2 │ ├── multi_node_example.yaml │ ├── multi_node_example1.yaml │ ├── multi_node_example2.yaml │ ├── multi_node_example_by_ddp.yaml │ ├── multi_node_example_by_deepspeed.yaml │ ├── zero2.json │ ├── zero2_npu.json │ ├── zero2_offload.json │ ├── zero3.json │ └── zero3_offload.json ├── eval.sh ├── gen_video.sh ├── rec_single_image.sh ├── rec_single_video.sh ├── save_hf_model.sh ├── test_var.sh ├── train_ddp.sh └── wfvae-large-16chn.json ├── requirements.txt ├── scripts ├── convert_weight.py ├── eval.py ├── expand_weight.py ├── merge_encoder_and_decoder.py ├── recon_single_image.py ├── recon_single_video.py ├── recon_video.py ├── save_hf_model.py └── vae_var_test.py ├── setup.py └── train_ddp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/README.md -------------------------------------------------------------------------------- /TRAIN_AND_VALIDATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/TRAIN_AND_VALIDATE.md -------------------------------------------------------------------------------- /assets/gt_5544.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/assets/gt_5544.jpg -------------------------------------------------------------------------------- /assets/logo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/assets/logo.jpg -------------------------------------------------------------------------------- /assets/report.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/assets/report.pdf -------------------------------------------------------------------------------- /causalvideovae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /causalvideovae/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /causalvideovae/dataset/ddp_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/dataset/ddp_sampler.py -------------------------------------------------------------------------------- /causalvideovae/dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/dataset/transform.py -------------------------------------------------------------------------------- /causalvideovae/dataset/video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/dataset/video_dataset.py -------------------------------------------------------------------------------- /causalvideovae/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /causalvideovae/eval/cal_fvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/cal_fvd.py -------------------------------------------------------------------------------- /causalvideovae/eval/cal_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/cal_lpips.py -------------------------------------------------------------------------------- /causalvideovae/eval/cal_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/cal_psnr.py -------------------------------------------------------------------------------- /causalvideovae/eval/cal_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/cal_ssim.py -------------------------------------------------------------------------------- /causalvideovae/eval/fvd/styleganv/fvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/fvd/styleganv/fvd.py -------------------------------------------------------------------------------- /causalvideovae/eval/fvd/videogpt/fvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/fvd/videogpt/fvd.py -------------------------------------------------------------------------------- /causalvideovae/eval/fvd/videogpt/pytorch_i3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/eval/fvd/videogpt/pytorch_i3d.py -------------------------------------------------------------------------------- /causalvideovae/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/__init__.py -------------------------------------------------------------------------------- /causalvideovae/model/configuration_videobase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/configuration_videobase.py -------------------------------------------------------------------------------- /causalvideovae/model/ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/ema_model.py -------------------------------------------------------------------------------- /causalvideovae/model/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/losses/__init__.py -------------------------------------------------------------------------------- /causalvideovae/model/losses/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/losses/discriminator.py -------------------------------------------------------------------------------- /causalvideovae/model/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/losses/lpips.py -------------------------------------------------------------------------------- /causalvideovae/model/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/losses/perceptual_loss.py -------------------------------------------------------------------------------- /causalvideovae/model/modeling_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modeling_output.py -------------------------------------------------------------------------------- /causalvideovae/model/modeling_videobase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modeling_videobase.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/__init__.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/attention.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/conv.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/normalize.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/ops.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/quant.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/resnet_block.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/updownsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/updownsample.py -------------------------------------------------------------------------------- /causalvideovae/model/modules/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/modules/wavelet.py -------------------------------------------------------------------------------- /causalvideovae/model/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/registry.py -------------------------------------------------------------------------------- /causalvideovae/model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /causalvideovae/model/utils/distrib_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/utils/distrib_utils.py -------------------------------------------------------------------------------- /causalvideovae/model/utils/module_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/utils/module_utils.py -------------------------------------------------------------------------------- /causalvideovae/model/utils/scheduler_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/utils/scheduler_utils.py -------------------------------------------------------------------------------- /causalvideovae/model/utils/wavelet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/utils/wavelet_utils.py -------------------------------------------------------------------------------- /causalvideovae/model/vae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/vae/__init__.py -------------------------------------------------------------------------------- /causalvideovae/model/vae/modeling_causalvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/vae/modeling_causalvae.py -------------------------------------------------------------------------------- /causalvideovae/model/vae/modeling_wfvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/vae/modeling_wfvae.py -------------------------------------------------------------------------------- /causalvideovae/model/vae/modeling_wfvae2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/model/vae/modeling_wfvae2.py -------------------------------------------------------------------------------- /causalvideovae/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /causalvideovae/utils/downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/utils/downloader.py -------------------------------------------------------------------------------- /causalvideovae/utils/taming_download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/utils/taming_download.py -------------------------------------------------------------------------------- /causalvideovae/utils/video_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/causalvideovae/utils/video_utils.py -------------------------------------------------------------------------------- /examples/accelerate_configs/ddp_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/ddp_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/debug.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/deepspeed_zero2_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/deepspeed_zero2_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/deepspeed_zero2_offload_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/deepspeed_zero2_offload_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/deepspeed_zero3_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/deepspeed_zero3_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/deepspeed_zero3_offload_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/deepspeed_zero3_offload_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/default_config.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/hostfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/hostfile -------------------------------------------------------------------------------- /examples/accelerate_configs/hostfile1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/hostfile1 -------------------------------------------------------------------------------- /examples/accelerate_configs/hostfile2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/hostfile2 -------------------------------------------------------------------------------- /examples/accelerate_configs/multi_node_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/multi_node_example.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/multi_node_example1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/multi_node_example1.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/multi_node_example2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/multi_node_example2.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/multi_node_example_by_ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/multi_node_example_by_ddp.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/multi_node_example_by_deepspeed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/multi_node_example_by_deepspeed.yaml -------------------------------------------------------------------------------- /examples/accelerate_configs/zero2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/zero2.json -------------------------------------------------------------------------------- /examples/accelerate_configs/zero2_npu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/zero2_npu.json -------------------------------------------------------------------------------- /examples/accelerate_configs/zero2_offload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/zero2_offload.json -------------------------------------------------------------------------------- /examples/accelerate_configs/zero3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/zero3.json -------------------------------------------------------------------------------- /examples/accelerate_configs/zero3_offload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/accelerate_configs/zero3_offload.json -------------------------------------------------------------------------------- /examples/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/eval.sh -------------------------------------------------------------------------------- /examples/gen_video.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/gen_video.sh -------------------------------------------------------------------------------- /examples/rec_single_image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/rec_single_image.sh -------------------------------------------------------------------------------- /examples/rec_single_video.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/rec_single_video.sh -------------------------------------------------------------------------------- /examples/save_hf_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/save_hf_model.sh -------------------------------------------------------------------------------- /examples/test_var.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/test_var.sh -------------------------------------------------------------------------------- /examples/train_ddp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/train_ddp.sh -------------------------------------------------------------------------------- /examples/wfvae-large-16chn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/examples/wfvae-large-16chn.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/convert_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/convert_weight.py -------------------------------------------------------------------------------- /scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/eval.py -------------------------------------------------------------------------------- /scripts/expand_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/expand_weight.py -------------------------------------------------------------------------------- /scripts/merge_encoder_and_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/merge_encoder_and_decoder.py -------------------------------------------------------------------------------- /scripts/recon_single_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/recon_single_image.py -------------------------------------------------------------------------------- /scripts/recon_single_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/recon_single_video.py -------------------------------------------------------------------------------- /scripts/recon_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/recon_video.py -------------------------------------------------------------------------------- /scripts/save_hf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/save_hf_model.py -------------------------------------------------------------------------------- /scripts/vae_var_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/scripts/vae_var_test.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/setup.py -------------------------------------------------------------------------------- /train_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-YuanGroup/WF-VAE/HEAD/train_ddp.py --------------------------------------------------------------------------------