├── .github └── ISSUE_TEMPLATE │ ├── PULL_REQUEST_TEMPLATE.md │ ├── bug_report.yaml │ └── feature-request.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── MODEL_LICENSE ├── README.md ├── README_ja.md ├── README_zh.md ├── finetune ├── README.md ├── README_ja.md ├── README_zh.md ├── accelerate_config.yaml ├── configs │ ├── zero2.yaml │ ├── zero2_offload.yaml │ ├── zero3.yaml │ └── zero3_offload.yaml ├── constants.py ├── datasets │ ├── __init__.py │ ├── bucket_sampler.py │ ├── i2v_dataset.py │ ├── t2v_dataset.py │ └── utils.py ├── models │ ├── __init__.py │ ├── cogvideox1_5_i2v │ │ ├── lora_trainer.py │ │ └── sft_trainer.py │ ├── cogvideox1_5_t2v │ │ ├── lora_trainer.py │ │ └── sft_trainer.py │ ├── cogvideox_i2v │ │ ├── lora_trainer.py │ │ └── sft_trainer.py │ ├── cogvideox_t2v │ │ ├── lora_trainer.py │ │ └── sft_trainer.py │ └── utils.py ├── schemas │ ├── __init__.py │ ├── args.py │ ├── components.py │ └── state.py ├── scripts │ └── extract_images.py ├── train.py ├── train_ddp_i2v.sh ├── train_ddp_t2v.sh ├── train_zero_i2v.sh ├── train_zero_t2v.sh ├── trainer.py └── utils │ ├── __init__.py │ ├── checkpointing.py │ ├── file_utils.py │ ├── memory_utils.py │ ├── optimizer_utils.py │ └── torch_utils.py ├── inference ├── cli_demo.py ├── cli_demo_quantization.py ├── cli_vae_demo.py ├── convert_demo.py ├── ddim_inversion.py ├── encoded.pt ├── gradio_composite_demo │ ├── README.md │ ├── app.py │ ├── example_images │ │ ├── beach.png │ │ ├── camping.png │ │ └── street.png │ ├── example_videos │ │ ├── horse.mp4 │ │ ├── kitten.mp4 │ │ └── train_running.mp4 │ ├── 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 ├── pyproject.toml ├── requirements.txt ├── resources ├── WECHAT.md ├── galary_prompt.md ├── hf_lora_weights.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 │ ├── cogvideox1.5_5b.yaml │ ├── cogvideox1.5_5b_i2v.yaml │ ├── cogvideox_2b.yaml │ ├── cogvideox_2b_lora.yaml │ ├── cogvideox_5b.yaml │ ├── cogvideox_5b_i2v.yaml │ ├── cogvideox_5b_i2v_lora.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-Caption-example.png │ └── cogvlm2-video-example.png ├── requirements.txt └── video_caption.py ├── convert_weight_deepspeed2hf.py ├── convert_weight_sat2hf.py ├── export_sat_lora_weight.py ├── llm_flux_cogvideox ├── generate.sh ├── gradio_page.py └── llm_flux_cogvideox.py ├── load_cogvideox_lora.py ├── parallel_inference ├── parallel_inference_xdit.py └── run.sh ├── replicate ├── cog.yaml ├── predict_i2v.py └── predict_t2v.py └── venhancer ├── README.md ├── README_ja.md └── README_zh.md /.github/ISSUE_TEMPLATE/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/.github/ISSUE_TEMPLATE/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/.github/ISSUE_TEMPLATE/bug_report.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/.github/ISSUE_TEMPLATE/feature-request.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/MODEL_LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/README.md -------------------------------------------------------------------------------- /README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/README_ja.md -------------------------------------------------------------------------------- /README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/README_zh.md -------------------------------------------------------------------------------- /finetune/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/README.md -------------------------------------------------------------------------------- /finetune/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/README_ja.md -------------------------------------------------------------------------------- /finetune/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/README_zh.md -------------------------------------------------------------------------------- /finetune/accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/accelerate_config.yaml -------------------------------------------------------------------------------- /finetune/configs/zero2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/configs/zero2.yaml -------------------------------------------------------------------------------- /finetune/configs/zero2_offload.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/configs/zero2_offload.yaml -------------------------------------------------------------------------------- /finetune/configs/zero3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/configs/zero3.yaml -------------------------------------------------------------------------------- /finetune/configs/zero3_offload.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/configs/zero3_offload.yaml -------------------------------------------------------------------------------- /finetune/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/constants.py -------------------------------------------------------------------------------- /finetune/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/datasets/__init__.py -------------------------------------------------------------------------------- /finetune/datasets/bucket_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/datasets/bucket_sampler.py -------------------------------------------------------------------------------- /finetune/datasets/i2v_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/datasets/i2v_dataset.py -------------------------------------------------------------------------------- /finetune/datasets/t2v_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/datasets/t2v_dataset.py -------------------------------------------------------------------------------- /finetune/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/datasets/utils.py -------------------------------------------------------------------------------- /finetune/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/__init__.py -------------------------------------------------------------------------------- /finetune/models/cogvideox1_5_i2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox1_5_i2v/lora_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox1_5_i2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox1_5_i2v/sft_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox1_5_t2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox1_5_t2v/lora_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox1_5_t2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox1_5_t2v/sft_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox_i2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox_i2v/lora_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox_i2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox_i2v/sft_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox_t2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox_t2v/lora_trainer.py -------------------------------------------------------------------------------- /finetune/models/cogvideox_t2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/cogvideox_t2v/sft_trainer.py -------------------------------------------------------------------------------- /finetune/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/models/utils.py -------------------------------------------------------------------------------- /finetune/schemas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/schemas/__init__.py -------------------------------------------------------------------------------- /finetune/schemas/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/schemas/args.py -------------------------------------------------------------------------------- /finetune/schemas/components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/schemas/components.py -------------------------------------------------------------------------------- /finetune/schemas/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/schemas/state.py -------------------------------------------------------------------------------- /finetune/scripts/extract_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/scripts/extract_images.py -------------------------------------------------------------------------------- /finetune/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/train.py -------------------------------------------------------------------------------- /finetune/train_ddp_i2v.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/train_ddp_i2v.sh -------------------------------------------------------------------------------- /finetune/train_ddp_t2v.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/train_ddp_t2v.sh -------------------------------------------------------------------------------- /finetune/train_zero_i2v.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/train_zero_i2v.sh -------------------------------------------------------------------------------- /finetune/train_zero_t2v.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/train_zero_t2v.sh -------------------------------------------------------------------------------- /finetune/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/trainer.py -------------------------------------------------------------------------------- /finetune/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/__init__.py -------------------------------------------------------------------------------- /finetune/utils/checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/checkpointing.py -------------------------------------------------------------------------------- /finetune/utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/file_utils.py -------------------------------------------------------------------------------- /finetune/utils/memory_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/memory_utils.py -------------------------------------------------------------------------------- /finetune/utils/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/optimizer_utils.py -------------------------------------------------------------------------------- /finetune/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/finetune/utils/torch_utils.py -------------------------------------------------------------------------------- /inference/cli_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/cli_demo.py -------------------------------------------------------------------------------- /inference/cli_demo_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/cli_demo_quantization.py -------------------------------------------------------------------------------- /inference/cli_vae_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/cli_vae_demo.py -------------------------------------------------------------------------------- /inference/convert_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/convert_demo.py -------------------------------------------------------------------------------- /inference/ddim_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/ddim_inversion.py -------------------------------------------------------------------------------- /inference/encoded.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/encoded.pt -------------------------------------------------------------------------------- /inference/gradio_composite_demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/README.md -------------------------------------------------------------------------------- /inference/gradio_composite_demo/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/app.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_images/beach.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_images/beach.png -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_images/camping.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_images/camping.png -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_images/street.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_images/street.png -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_videos/horse.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_videos/horse.mp4 -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_videos/kitten.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_videos/kitten.mp4 -------------------------------------------------------------------------------- /inference/gradio_composite_demo/example_videos/train_running.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/example_videos/train_running.mp4 -------------------------------------------------------------------------------- /inference/gradio_composite_demo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/requirements.txt -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_2R.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_2R.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_HDv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_HDv3.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/IFNet_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/IFNet_m.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/RIFE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/RIFE.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/RIFE_HDv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/laplacian.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/loss.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/pytorch_msssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/pytorch_msssim/__init__.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/refine.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/refine_2R.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/refine_2R.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife/warplayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife/warplayer.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/rife_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/rife_model.py -------------------------------------------------------------------------------- /inference/gradio_composite_demo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_composite_demo/utils.py -------------------------------------------------------------------------------- /inference/gradio_web_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/inference/gradio_web_demo.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/WECHAT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/WECHAT.md -------------------------------------------------------------------------------- /resources/galary_prompt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/galary_prompt.md -------------------------------------------------------------------------------- /resources/hf_lora_weights.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/hf_lora_weights.png -------------------------------------------------------------------------------- /resources/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/logo.svg -------------------------------------------------------------------------------- /resources/videos/1.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/videos/1.mp4 -------------------------------------------------------------------------------- /resources/videos/2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/videos/2.mp4 -------------------------------------------------------------------------------- /resources/videos/3.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/videos/3.mp4 -------------------------------------------------------------------------------- /resources/videos/4.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/videos/4.mp4 -------------------------------------------------------------------------------- /resources/web_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/web_demo.png -------------------------------------------------------------------------------- /resources/wechat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/resources/wechat.jpg -------------------------------------------------------------------------------- /sat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/README.md -------------------------------------------------------------------------------- /sat/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/README_ja.md -------------------------------------------------------------------------------- /sat/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/README_zh.md -------------------------------------------------------------------------------- /sat/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/arguments.py -------------------------------------------------------------------------------- /sat/configs/cogvideox1.5_5b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox1.5_5b.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox1.5_5b_i2v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox1.5_5b_i2v.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_2b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_2b.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_2b_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_2b_lora.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_5b.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b_i2v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_5b_i2v.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b_i2v_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_5b_i2v_lora.yaml -------------------------------------------------------------------------------- /sat/configs/cogvideox_5b_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/cogvideox_5b_lora.yaml -------------------------------------------------------------------------------- /sat/configs/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/inference.yaml -------------------------------------------------------------------------------- /sat/configs/sft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/sft.yaml -------------------------------------------------------------------------------- /sat/configs/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/configs/test.txt -------------------------------------------------------------------------------- /sat/data_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/data_video.py -------------------------------------------------------------------------------- /sat/diffusion_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/diffusion_video.py -------------------------------------------------------------------------------- /sat/dit_video_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/dit_video_concat.py -------------------------------------------------------------------------------- /sat/finetune_multi_gpus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/finetune_multi_gpus.sh -------------------------------------------------------------------------------- /sat/finetune_single_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/finetune_single_gpu.sh -------------------------------------------------------------------------------- /sat/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/inference.sh -------------------------------------------------------------------------------- /sat/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/requirements.txt -------------------------------------------------------------------------------- /sat/sample_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sample_video.py -------------------------------------------------------------------------------- /sat/sgm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/__init__.py -------------------------------------------------------------------------------- /sat/sgm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/lr_scheduler.py -------------------------------------------------------------------------------- /sat/sgm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/models/__init__.py -------------------------------------------------------------------------------- /sat/sgm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/models/autoencoder.py -------------------------------------------------------------------------------- /sat/sgm/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/discriminator_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/discriminator_loss.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/losses/lpips.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/losses/video_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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 2 | -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/loss/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/loss/lpips.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/model/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/model/model.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/util.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/lpips/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/lpips/vqperceptual.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/magvit2_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/magvit2_pytorch.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/base.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/regularizers/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/regularizers/quantize.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/temporal_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/temporal_ae.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_dec_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/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/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/movq_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/movq_modules.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/quantize.py -------------------------------------------------------------------------------- /sat/sgm/modules/autoencoding/vqvae/vqvae_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/autoencoding/vqvae/vqvae_blocks.py -------------------------------------------------------------------------------- /sat/sgm/modules/cp_enc_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/cp_enc_dec.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/__init__.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser_scaling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/denoiser_weighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/denoiser_weighting.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/discretizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/discretizer.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/guiders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/guiders.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/lora.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/loss.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sampling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sampling_utils.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/sigma_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/sigma_sampling.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /sat/sgm/modules/diffusionmodules/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/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/zai-org/CogVideo/HEAD/sat/sgm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /sat/sgm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/ema.py -------------------------------------------------------------------------------- /sat/sgm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sat/sgm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/encoders/modules.py -------------------------------------------------------------------------------- /sat/sgm/modules/video_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/modules/video_attention.py -------------------------------------------------------------------------------- /sat/sgm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/util.py -------------------------------------------------------------------------------- /sat/sgm/webds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/sgm/webds.py -------------------------------------------------------------------------------- /sat/train_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/train_video.py -------------------------------------------------------------------------------- /sat/vae_modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/attention.py -------------------------------------------------------------------------------- /sat/vae_modules/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/autoencoder.py -------------------------------------------------------------------------------- /sat/vae_modules/cp_enc_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/cp_enc_dec.py -------------------------------------------------------------------------------- /sat/vae_modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/ema.py -------------------------------------------------------------------------------- /sat/vae_modules/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/regularizers.py -------------------------------------------------------------------------------- /sat/vae_modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/sat/vae_modules/utils.py -------------------------------------------------------------------------------- /tools/caption/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/README.md -------------------------------------------------------------------------------- /tools/caption/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/README_ja.md -------------------------------------------------------------------------------- /tools/caption/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/README_zh.md -------------------------------------------------------------------------------- /tools/caption/assests/CogVLM2-Caption-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/assests/CogVLM2-Caption-example.png -------------------------------------------------------------------------------- /tools/caption/assests/cogvlm2-video-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/assests/cogvlm2-video-example.png -------------------------------------------------------------------------------- /tools/caption/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/requirements.txt -------------------------------------------------------------------------------- /tools/caption/video_caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/caption/video_caption.py -------------------------------------------------------------------------------- /tools/convert_weight_deepspeed2hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/convert_weight_deepspeed2hf.py -------------------------------------------------------------------------------- /tools/convert_weight_sat2hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/convert_weight_sat2hf.py -------------------------------------------------------------------------------- /tools/export_sat_lora_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/export_sat_lora_weight.py -------------------------------------------------------------------------------- /tools/llm_flux_cogvideox/generate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/llm_flux_cogvideox/generate.sh -------------------------------------------------------------------------------- /tools/llm_flux_cogvideox/gradio_page.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/llm_flux_cogvideox/gradio_page.py -------------------------------------------------------------------------------- /tools/llm_flux_cogvideox/llm_flux_cogvideox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/llm_flux_cogvideox/llm_flux_cogvideox.py -------------------------------------------------------------------------------- /tools/load_cogvideox_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/load_cogvideox_lora.py -------------------------------------------------------------------------------- /tools/parallel_inference/parallel_inference_xdit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/parallel_inference/parallel_inference_xdit.py -------------------------------------------------------------------------------- /tools/parallel_inference/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/parallel_inference/run.sh -------------------------------------------------------------------------------- /tools/replicate/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/replicate/cog.yaml -------------------------------------------------------------------------------- /tools/replicate/predict_i2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/replicate/predict_i2v.py -------------------------------------------------------------------------------- /tools/replicate/predict_t2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/replicate/predict_t2v.py -------------------------------------------------------------------------------- /tools/venhancer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/venhancer/README.md -------------------------------------------------------------------------------- /tools/venhancer/README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/venhancer/README_ja.md -------------------------------------------------------------------------------- /tools/venhancer/README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogVideo/HEAD/tools/venhancer/README_zh.md --------------------------------------------------------------------------------