├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.yaml │ └── feature-request.yaml └── PULL_REQUEST_TEMPLATE │ └── pr_template.md ├── .gitignore ├── LICENSE ├── MODEL_LICENSE ├── README.md ├── README_ja.md ├── README_zh.md ├── inference ├── cli_demo.py ├── cli_demo_quantization.py ├── cli_vae_demo.py ├── convert_demo.py ├── encoded.pt ├── gradio_composite_demo │ ├── README.md │ ├── app.py │ ├── requirements.txt │ ├── rife │ │ ├── IFNet.py │ │ ├── IFNet_2R.py │ │ ├── IFNet_HDv3.py │ │ ├── IFNet_m.py │ │ ├── RIFE.py │ │ ├── RIFE_HDv3.py │ │ ├── __init__.py │ │ ├── laplacian.py │ │ ├── loss.py │ │ ├── pytorch_msssim │ │ │ └── __init__.py │ │ ├── refine.py │ │ ├── refine_2R.py │ │ └── warplayer.py │ ├── rife_model.py │ └── utils.py ├── gradio_web_demo.py └── streamlit_web_demo.py ├── pyproject.toml ├── requirements.txt ├── resources ├── WECHAT.md ├── contribute.md ├── contribute_zh.md ├── galary_prompt.md ├── gradio_demo.png ├── logo.svg ├── videos │ ├── 1.mp4 │ ├── 2.mp4 │ ├── 3.mp4 │ └── 4.mp4 ├── web_demo.png └── wechat.jpg ├── sat ├── README.md ├── README_ja.md ├── README_zh.md ├── arguments.py ├── configs │ ├── cogvideox_2b.yaml │ ├── cogvideox_2b_lora.yaml │ ├── cogvideox_5b.yaml │ ├── cogvideox_5b_lora.yaml │ ├── inference.yaml │ ├── sft.yaml │ └── test.txt ├── data_video.py ├── diffusion_video.py ├── dit_video_concat.py ├── finetune_multi_gpus.sh ├── finetune_single_gpu.sh ├── inference.sh ├── requirements.txt ├── sample_video.py ├── sgm │ ├── __init__.py │ ├── lr_scheduler.py │ ├── models │ │ ├── __init__.py │ │ └── autoencoder.py │ ├── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── autoencoding │ │ │ ├── __init__.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── discriminator_loss.py │ │ │ │ ├── lpips.py │ │ │ │ └── video_loss.py │ │ │ ├── lpips │ │ │ │ ├── __init__.py │ │ │ │ ├── loss │ │ │ │ │ ├── .gitignore │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── lpips.py │ │ │ │ ├── model │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── model.py │ │ │ │ ├── util.py │ │ │ │ └── vqperceptual.py │ │ │ ├── magvit2_pytorch.py │ │ │ ├── regularizers │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── finite_scalar_quantization.py │ │ │ │ ├── lookup_free_quantization.py │ │ │ │ └── quantize.py │ │ │ ├── temporal_ae.py │ │ │ └── vqvae │ │ │ │ ├── movq_dec_3d.py │ │ │ │ ├── movq_dec_3d_dev.py │ │ │ │ ├── movq_enc_3d.py │ │ │ │ ├── movq_modules.py │ │ │ │ ├── quantize.py │ │ │ │ └── vqvae_blocks.py │ │ ├── cp_enc_dec.py │ │ ├── diffusionmodules │ │ │ ├── __init__.py │ │ │ ├── denoiser.py │ │ │ ├── denoiser_scaling.py │ │ │ ├── denoiser_weighting.py │ │ │ ├── discretizer.py │ │ │ ├── guiders.py │ │ │ ├── lora.py │ │ │ ├── loss.py │ │ │ ├── model.py │ │ │ ├── openaimodel.py │ │ │ ├── sampling.py │ │ │ ├── sampling_utils.py │ │ │ ├── sigma_sampling.py │ │ │ ├── util.py │ │ │ └── wrappers.py │ │ ├── distributions │ │ │ ├── __init__.py │ │ │ └── distributions.py │ │ ├── ema.py │ │ ├── encoders │ │ │ ├── __init__.py │ │ │ └── modules.py │ │ └── video_attention.py │ ├── util.py │ └── webds.py ├── train_video.py └── vae_modules │ ├── attention.py │ ├── autoencoder.py │ ├── cp_enc_dec.py │ ├── ema.py │ ├── regularizers.py │ └── utils.py └── tools ├── caption ├── README.md ├── README_ja.md ├── README_zh.md └── assests │ └── cogvlm2-video-example.png ├── convert_weight_sat2hf.py └── venhancer ├── README.md ├── README_ja.md └── README_zh.md /.github/ISSUE_TEMPLATE/bug_report.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/.github/ISSUE_TEMPLATE/bug_report.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/.github/ISSUE_TEMPLATE/feature-request.yaml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE/pr_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/.github/PULL_REQUEST_TEMPLATE/pr_template.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/MODEL_LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/README.md -------------------------------------------------------------------------------- /README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/README_ja.md -------------------------------------------------------------------------------- /README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/README_zh.md -------------------------------------------------------------------------------- /inference/cli_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/cli_demo.py -------------------------------------------------------------------------------- /inference/cli_demo_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/cli_demo_quantization.py -------------------------------------------------------------------------------- /inference/cli_vae_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/cli_vae_demo.py -------------------------------------------------------------------------------- /inference/convert_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/convert_demo.py -------------------------------------------------------------------------------- /inference/encoded.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/encoded.pt -------------------------------------------------------------------------------- /inference/gradio_composite_demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/README.md -------------------------------------------------------------------------------- /inference/gradio_composite_demo/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/app.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/requirements.txt -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_2R.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_2R.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_HDv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_HDv3.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_m.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/RIFE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/RIFE.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/RIFE_HDv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/RIFE_HDv3.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/laplacian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/laplacian.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/loss.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/pytorch_msssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/pytorch_msssim/__init__.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/refine.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/refine_2R.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/refine_2R.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/warplayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife/warplayer.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/rife_model.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_composite_demo/utils.py -------------------------------------------------------------------------------- /inference/gradio_web_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/gradio_web_demo.py -------------------------------------------------------------------------------- /inference/streamlit_web_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/inference/streamlit_web_demo.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/WECHAT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/WECHAT.md -------------------------------------------------------------------------------- /resources/contribute.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/contribute.md -------------------------------------------------------------------------------- /resources/contribute_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/contribute_zh.md -------------------------------------------------------------------------------- /resources/galary_prompt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/galary_prompt.md -------------------------------------------------------------------------------- /resources/gradio_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/gradio_demo.png -------------------------------------------------------------------------------- /resources/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/logo.svg -------------------------------------------------------------------------------- /resources/videos/1.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/videos/1.mp4 -------------------------------------------------------------------------------- /resources/videos/2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/videos/2.mp4 -------------------------------------------------------------------------------- /resources/videos/3.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/videos/3.mp4 -------------------------------------------------------------------------------- /resources/videos/4.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/videos/4.mp4 -------------------------------------------------------------------------------- /resources/web_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/web_demo.png -------------------------------------------------------------------------------- /resources/wechat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/resources/wechat.jpg -------------------------------------------------------------------------------- /sat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/README.md -------------------------------------------------------------------------------- /sat/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/README_ja.md -------------------------------------------------------------------------------- /sat/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/README_zh.md -------------------------------------------------------------------------------- /sat/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/arguments.py -------------------------------------------------------------------------------- /sat/configs/cogvideox_2b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/cogvideox_2b.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_2b_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/cogvideox_2b_lora.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/cogvideox_5b.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/cogvideox_5b_lora.yaml -------------------------------------------------------------------------------- /sat/configs/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/inference.yaml -------------------------------------------------------------------------------- /sat/configs/sft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/sft.yaml -------------------------------------------------------------------------------- /sat/configs/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/configs/test.txt -------------------------------------------------------------------------------- /sat/data_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/data_video.py -------------------------------------------------------------------------------- /sat/diffusion_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/diffusion_video.py -------------------------------------------------------------------------------- /sat/dit_video_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/dit_video_concat.py -------------------------------------------------------------------------------- /sat/finetune_multi_gpus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/finetune_multi_gpus.sh -------------------------------------------------------------------------------- /sat/finetune_single_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/finetune_single_gpu.sh -------------------------------------------------------------------------------- /sat/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/inference.sh -------------------------------------------------------------------------------- /sat/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/requirements.txt -------------------------------------------------------------------------------- /sat/sample_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sample_video.py -------------------------------------------------------------------------------- /sat/sgm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/__init__.py -------------------------------------------------------------------------------- /sat/sgm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/lr_scheduler.py -------------------------------------------------------------------------------- /sat/sgm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/models/__init__.py -------------------------------------------------------------------------------- /sat/sgm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/models/autoencoder.py -------------------------------------------------------------------------------- /sat/sgm/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/attention.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/discriminator_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/discriminator_loss.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/lpips.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/video_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/video_loss.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/loss/.gitignore: -------------------------------------------------------------------------------- 1 | vgg.pth -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/loss/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/loss/LICENSE -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/loss/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/loss/lpips.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/model/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/model/LICENSE -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/model/model.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/util.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/vqperceptual.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/magvit2_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/magvit2_pytorch.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/base.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/quantize.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/temporal_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/temporal_ae.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_dec_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_dec_3d.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_dec_3d_dev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_dec_3d_dev.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_modules.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/quantize.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/vqvae_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/vqvae_blocks.py -------------------------------------------------------------------------------- /sat/sgm/modules/cp_enc_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/cp_enc_dec.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser_scaling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser_weighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser_weighting.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/discretizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/discretizer.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/guiders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/guiders.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/lora.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/loss.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sampling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sampling_utils.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sigma_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sigma_sampling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/wrappers.py -------------------------------------------------------------------------------- /sat/sgm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /sat/sgm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/ema.py -------------------------------------------------------------------------------- /sat/sgm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/encoders/modules.py -------------------------------------------------------------------------------- /sat/sgm/modules/video_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/modules/video_attention.py -------------------------------------------------------------------------------- /sat/sgm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/util.py -------------------------------------------------------------------------------- /sat/sgm/webds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/sgm/webds.py -------------------------------------------------------------------------------- /sat/train_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/train_video.py -------------------------------------------------------------------------------- /sat/vae_modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/attention.py -------------------------------------------------------------------------------- /sat/vae_modules/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/autoencoder.py -------------------------------------------------------------------------------- /sat/vae_modules/cp_enc_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/cp_enc_dec.py -------------------------------------------------------------------------------- /sat/vae_modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/ema.py -------------------------------------------------------------------------------- /sat/vae_modules/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/regularizers.py -------------------------------------------------------------------------------- /sat/vae_modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/sat/vae_modules/utils.py -------------------------------------------------------------------------------- /tools/caption/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/caption/README.md -------------------------------------------------------------------------------- /tools/caption/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/caption/README_ja.md -------------------------------------------------------------------------------- /tools/caption/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/caption/README_zh.md -------------------------------------------------------------------------------- /tools/caption/assests/cogvlm2-video-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/caption/assests/cogvlm2-video-example.png -------------------------------------------------------------------------------- /tools/convert_weight_sat2hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/convert_weight_sat2hf.py -------------------------------------------------------------------------------- /tools/venhancer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/venhancer/README.md -------------------------------------------------------------------------------- /tools/venhancer/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/venhancer/README_ja.md -------------------------------------------------------------------------------- /tools/venhancer/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/peanutcocktail/CogVideo/HEAD/tools/venhancer/README_zh.md --------------------------------------------------------------------------------