├── .env.template ├── .github ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── python-lint.yml │ ├── test-webdoc-deploy.yml │ └── webdoc-deploy.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .python-version ├── LICENSE ├── README.md ├── docker └── Dockerfile ├── docs ├── 01-Intro.md ├── 02-Installation.md ├── 03-Inference │ ├── 01-CLI.md │ └── 02-API.md ├── 04-Finetune │ ├── 01-Prerequisites.mdx │ ├── 02-Quick Start.md │ └── 03-Data Format.md └── 05-Model Card.mdx ├── gradio ├── configs │ ├── t2i.yaml │ └── t2v.yaml ├── gradio_infer_demo.py ├── gradio_lora_demo.py ├── gradio_ui.py ├── styles │ └── mono.py └── utils │ ├── __init__.py │ ├── io.py │ ├── logging.py │ ├── misc.py │ └── task.py ├── pdm.lock ├── pyproject.toml ├── quickstart ├── data │ ├── i2v │ │ ├── test │ │ │ ├── 6619a387964ad49eb1127079d25d17d8.png │ │ │ ├── 8ae679ab483ab344c881d4a813e0cb51.png │ │ │ ├── a3c275fc2eb0a67168a7c58a6a9adb14.png │ │ │ └── metadata.jsonl │ │ └── train │ │ │ ├── images │ │ │ ├── 0bb5f6dbf8ed2e0060f0ac4164b24847.png │ │ │ ├── 1d50a3d9703f152758d5422c8b48010f.png │ │ │ ├── 2c1ed5408882479b06681f7cf372916a.png │ │ │ ├── 3f0979e6cae25447f416372c49ad5e07.png │ │ │ ├── 4adbb3a2945c9edd78785daccfd23e80.png │ │ │ └── metadata.jsonl │ │ │ ├── metadata.jsonl │ │ │ └── videos │ │ │ ├── 0bb5f6dbf8ed2e0060f0ac4164b24847.mp4 │ │ │ ├── 1d50a3d9703f152758d5422c8b48010f.mp4 │ │ │ ├── 2c1ed5408882479b06681f7cf372916a.mp4 │ │ │ ├── 3f0979e6cae25447f416372c49ad5e07.mp4 │ │ │ ├── 4adbb3a2945c9edd78785daccfd23e80.mp4 │ │ │ └── metadata.jsonl │ ├── t2i │ │ ├── test │ │ │ └── prompt.jsonl │ │ └── train │ │ │ ├── 0bb5f6dbf8ed2e0060f0ac4164b24847.png │ │ │ ├── 1d50a3d9703f152758d5422c8b48010f.png │ │ │ ├── 2c1ed5408882479b06681f7cf372916a.png │ │ │ ├── 3f0979e6cae25447f416372c49ad5e07.png │ │ │ ├── 4adbb3a2945c9edd78785daccfd23e80.png │ │ │ └── metadata.jsonl │ └── t2v │ │ ├── test │ │ └── prompt.jsonl │ │ └── train │ │ ├── 0bb5f6dbf8ed2e0060f0ac4164b24847.mp4 │ │ ├── 1d50a3d9703f152758d5422c8b48010f.mp4 │ │ ├── 2c1ed5408882479b06681f7cf372916a.mp4 │ │ ├── 3f0979e6cae25447f416372c49ad5e07.mp4 │ │ ├── 4adbb3a2945c9edd78785daccfd23e80.mp4 │ │ └── metadata.jsonl └── scripts │ ├── i2v │ ├── config.yaml │ └── start_train.sh │ ├── t2i │ ├── config.yaml │ └── start_train.sh │ ├── t2v │ ├── config.yaml │ └── start_train.sh │ └── train.py ├── src └── cogkit │ ├── __init__.py │ ├── __main__.py │ ├── api │ ├── __init__.py │ ├── application.py │ ├── dependencies.py │ ├── logging.py │ ├── models │ │ ├── __init__.py │ │ ├── images │ │ │ ├── __init__.py │ │ │ ├── generation_params.py │ │ │ └── response.py │ │ ├── request.py │ │ └── response.py │ ├── routers │ │ ├── __init__.py │ │ └── images.py │ ├── services │ │ ├── __init__.py │ │ └── image_generation.py │ ├── settings.py │ └── state.py │ ├── cli │ ├── __init__.py │ ├── inference.py │ └── launch.py │ ├── finetune │ ├── __init__.py │ ├── _register.py │ ├── base │ │ ├── __init__.py │ │ ├── base_args.py │ │ ├── base_component.py │ │ ├── base_state.py │ │ └── base_trainer.py │ ├── datasets │ │ ├── __init__.py │ │ ├── i2v_dataset.py │ │ ├── t2i_dataset.py │ │ ├── t2v_dataset.py │ │ └── utils.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── cogvideo │ │ │ │ ├── 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 │ │ │ └── cogview │ │ │ │ └── cogview4 │ │ │ │ ├── lora_trainer.py │ │ │ │ ├── lora_trainer_packing.py │ │ │ │ ├── sft_trainer.py │ │ │ │ └── sft_trainer_packing.py │ │ ├── schemas │ │ │ ├── __init__.py │ │ │ ├── args.py │ │ │ ├── components.py │ │ │ └── state.py │ │ └── trainer.py │ ├── llm │ │ └── models │ │ │ └── __init__.py │ ├── logger.py │ ├── samplers │ │ ├── __init__.py │ │ └── packing_sampler.py │ └── utils │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── ckpt.py │ │ ├── dist.py │ │ ├── filters.py │ │ ├── io.py │ │ ├── memory.py │ │ ├── misc.py │ │ └── tracker.py │ ├── logging.py │ ├── python │ ├── __init__.py │ └── generation │ │ ├── image.py │ │ ├── util.py │ │ └── video.py │ ├── types │ ├── __init__.py │ └── generation_mode.py │ └── utils │ ├── __init__.py │ ├── diffusion_pipeline.py │ ├── dtype.py │ ├── load.py │ ├── lora.py │ ├── misc.py │ ├── path.py │ ├── prompt.py │ ├── random.py │ └── seed.py ├── tests ├── __init__.py ├── test_expand_list.py ├── test_mean_filter.py ├── test_sampler.py └── utils │ └── test_path.py └── tools ├── converters ├── merge.py └── zero2diffusers.py ├── pixelart_dataset └── download_convert.py └── resize_img.py /.env.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.env.template -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/python-lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.github/workflows/python-lint.yml -------------------------------------------------------------------------------- /.github/workflows/test-webdoc-deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.github/workflows/test-webdoc-deploy.yml -------------------------------------------------------------------------------- /.github/workflows/webdoc-deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.github/workflows/webdoc-deploy.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.10 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/README.md -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docs/01-Intro.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/01-Intro.md -------------------------------------------------------------------------------- /docs/02-Installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/02-Installation.md -------------------------------------------------------------------------------- /docs/03-Inference/01-CLI.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/03-Inference/01-CLI.md -------------------------------------------------------------------------------- /docs/03-Inference/02-API.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/03-Inference/02-API.md -------------------------------------------------------------------------------- /docs/04-Finetune/01-Prerequisites.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/04-Finetune/01-Prerequisites.mdx -------------------------------------------------------------------------------- /docs/04-Finetune/02-Quick Start.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/04-Finetune/02-Quick Start.md -------------------------------------------------------------------------------- /docs/04-Finetune/03-Data Format.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/04-Finetune/03-Data Format.md -------------------------------------------------------------------------------- /docs/05-Model Card.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/docs/05-Model Card.mdx -------------------------------------------------------------------------------- /gradio/configs/t2i.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/configs/t2i.yaml -------------------------------------------------------------------------------- /gradio/configs/t2v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/configs/t2v.yaml -------------------------------------------------------------------------------- /gradio/gradio_infer_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/gradio_infer_demo.py -------------------------------------------------------------------------------- /gradio/gradio_lora_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/gradio_lora_demo.py -------------------------------------------------------------------------------- /gradio/gradio_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/gradio_ui.py -------------------------------------------------------------------------------- /gradio/styles/mono.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/styles/mono.py -------------------------------------------------------------------------------- /gradio/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/utils/__init__.py -------------------------------------------------------------------------------- /gradio/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/utils/io.py -------------------------------------------------------------------------------- /gradio/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/utils/logging.py -------------------------------------------------------------------------------- /gradio/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/utils/misc.py -------------------------------------------------------------------------------- /gradio/utils/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/gradio/utils/task.py -------------------------------------------------------------------------------- /pdm.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/pdm.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/pyproject.toml -------------------------------------------------------------------------------- /quickstart/data/i2v/test/6619a387964ad49eb1127079d25d17d8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/test/6619a387964ad49eb1127079d25d17d8.png -------------------------------------------------------------------------------- /quickstart/data/i2v/test/8ae679ab483ab344c881d4a813e0cb51.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/test/8ae679ab483ab344c881d4a813e0cb51.png -------------------------------------------------------------------------------- /quickstart/data/i2v/test/a3c275fc2eb0a67168a7c58a6a9adb14.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/test/a3c275fc2eb0a67168a7c58a6a9adb14.png -------------------------------------------------------------------------------- /quickstart/data/i2v/test/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/test/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/0bb5f6dbf8ed2e0060f0ac4164b24847.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/0bb5f6dbf8ed2e0060f0ac4164b24847.png -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/1d50a3d9703f152758d5422c8b48010f.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/1d50a3d9703f152758d5422c8b48010f.png -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/2c1ed5408882479b06681f7cf372916a.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/2c1ed5408882479b06681f7cf372916a.png -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/3f0979e6cae25447f416372c49ad5e07.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/3f0979e6cae25447f416372c49ad5e07.png -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/4adbb3a2945c9edd78785daccfd23e80.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/4adbb3a2945c9edd78785daccfd23e80.png -------------------------------------------------------------------------------- /quickstart/data/i2v/train/images/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/images/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/data/i2v/train/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/0bb5f6dbf8ed2e0060f0ac4164b24847.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/0bb5f6dbf8ed2e0060f0ac4164b24847.mp4 -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/1d50a3d9703f152758d5422c8b48010f.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/1d50a3d9703f152758d5422c8b48010f.mp4 -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/2c1ed5408882479b06681f7cf372916a.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/2c1ed5408882479b06681f7cf372916a.mp4 -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/3f0979e6cae25447f416372c49ad5e07.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/3f0979e6cae25447f416372c49ad5e07.mp4 -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/4adbb3a2945c9edd78785daccfd23e80.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/4adbb3a2945c9edd78785daccfd23e80.mp4 -------------------------------------------------------------------------------- /quickstart/data/i2v/train/videos/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/i2v/train/videos/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/data/t2i/test/prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/test/prompt.jsonl -------------------------------------------------------------------------------- /quickstart/data/t2i/train/0bb5f6dbf8ed2e0060f0ac4164b24847.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/0bb5f6dbf8ed2e0060f0ac4164b24847.png -------------------------------------------------------------------------------- /quickstart/data/t2i/train/1d50a3d9703f152758d5422c8b48010f.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/1d50a3d9703f152758d5422c8b48010f.png -------------------------------------------------------------------------------- /quickstart/data/t2i/train/2c1ed5408882479b06681f7cf372916a.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/2c1ed5408882479b06681f7cf372916a.png -------------------------------------------------------------------------------- /quickstart/data/t2i/train/3f0979e6cae25447f416372c49ad5e07.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/3f0979e6cae25447f416372c49ad5e07.png -------------------------------------------------------------------------------- /quickstart/data/t2i/train/4adbb3a2945c9edd78785daccfd23e80.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/4adbb3a2945c9edd78785daccfd23e80.png -------------------------------------------------------------------------------- /quickstart/data/t2i/train/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2i/train/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/data/t2v/test/prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/test/prompt.jsonl -------------------------------------------------------------------------------- /quickstart/data/t2v/train/0bb5f6dbf8ed2e0060f0ac4164b24847.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/0bb5f6dbf8ed2e0060f0ac4164b24847.mp4 -------------------------------------------------------------------------------- /quickstart/data/t2v/train/1d50a3d9703f152758d5422c8b48010f.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/1d50a3d9703f152758d5422c8b48010f.mp4 -------------------------------------------------------------------------------- /quickstart/data/t2v/train/2c1ed5408882479b06681f7cf372916a.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/2c1ed5408882479b06681f7cf372916a.mp4 -------------------------------------------------------------------------------- /quickstart/data/t2v/train/3f0979e6cae25447f416372c49ad5e07.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/3f0979e6cae25447f416372c49ad5e07.mp4 -------------------------------------------------------------------------------- /quickstart/data/t2v/train/4adbb3a2945c9edd78785daccfd23e80.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/4adbb3a2945c9edd78785daccfd23e80.mp4 -------------------------------------------------------------------------------- /quickstart/data/t2v/train/metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/data/t2v/train/metadata.jsonl -------------------------------------------------------------------------------- /quickstart/scripts/i2v/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/i2v/config.yaml -------------------------------------------------------------------------------- /quickstart/scripts/i2v/start_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/i2v/start_train.sh -------------------------------------------------------------------------------- /quickstart/scripts/t2i/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/t2i/config.yaml -------------------------------------------------------------------------------- /quickstart/scripts/t2i/start_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/t2i/start_train.sh -------------------------------------------------------------------------------- /quickstart/scripts/t2v/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/t2v/config.yaml -------------------------------------------------------------------------------- /quickstart/scripts/t2v/start_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/t2v/start_train.sh -------------------------------------------------------------------------------- /quickstart/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/quickstart/scripts/train.py -------------------------------------------------------------------------------- /src/cogkit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/__init__.py -------------------------------------------------------------------------------- /src/cogkit/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/__main__.py -------------------------------------------------------------------------------- /src/cogkit/api/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/__init__.py -------------------------------------------------------------------------------- /src/cogkit/api/application.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/application.py -------------------------------------------------------------------------------- /src/cogkit/api/dependencies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/dependencies.py -------------------------------------------------------------------------------- /src/cogkit/api/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/logging.py -------------------------------------------------------------------------------- /src/cogkit/api/models/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /src/cogkit/api/models/images/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/models/images/__init__.py -------------------------------------------------------------------------------- /src/cogkit/api/models/images/generation_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/models/images/generation_params.py -------------------------------------------------------------------------------- /src/cogkit/api/models/images/response.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/models/images/response.py -------------------------------------------------------------------------------- /src/cogkit/api/models/request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/models/request.py -------------------------------------------------------------------------------- /src/cogkit/api/models/response.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/models/response.py -------------------------------------------------------------------------------- /src/cogkit/api/routers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/routers/__init__.py -------------------------------------------------------------------------------- /src/cogkit/api/routers/images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/routers/images.py -------------------------------------------------------------------------------- /src/cogkit/api/services/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/services/__init__.py -------------------------------------------------------------------------------- /src/cogkit/api/services/image_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/services/image_generation.py -------------------------------------------------------------------------------- /src/cogkit/api/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/settings.py -------------------------------------------------------------------------------- /src/cogkit/api/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/api/state.py -------------------------------------------------------------------------------- /src/cogkit/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/cli/__init__.py -------------------------------------------------------------------------------- /src/cogkit/cli/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/cli/inference.py -------------------------------------------------------------------------------- /src/cogkit/cli/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/cli/launch.py -------------------------------------------------------------------------------- /src/cogkit/finetune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/_register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/_register.py -------------------------------------------------------------------------------- /src/cogkit/finetune/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/base/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/base/base_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/base/base_args.py -------------------------------------------------------------------------------- /src/cogkit/finetune/base/base_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/base/base_component.py -------------------------------------------------------------------------------- /src/cogkit/finetune/base/base_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/base/base_state.py -------------------------------------------------------------------------------- /src/cogkit/finetune/base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/base/base_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/datasets/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/datasets/i2v_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/datasets/i2v_dataset.py -------------------------------------------------------------------------------- /src/cogkit/finetune/datasets/t2i_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/datasets/t2i_dataset.py -------------------------------------------------------------------------------- /src/cogkit/finetune/datasets/t2v_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/datasets/t2v_dataset.py -------------------------------------------------------------------------------- /src/cogkit/finetune/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/datasets/utils.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_i2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_i2v/lora_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_i2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_i2v/sft_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_t2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_t2v/lora_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_t2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox1_5_t2v/sft_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_i2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_i2v/lora_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_i2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_i2v/sft_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_t2v/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_t2v/lora_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_t2v/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogvideo/cogvideox_t2v/sft_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogview/cogview4/lora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogview/cogview4/lora_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogview/cogview4/lora_trainer_packing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogview/cogview4/lora_trainer_packing.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogview/cogview4/sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogview/cogview4/sft_trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/models/cogview/cogview4/sft_trainer_packing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/models/cogview/cogview4/sft_trainer_packing.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/schemas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/schemas/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/schemas/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/schemas/args.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/schemas/components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/schemas/components.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/schemas/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/schemas/state.py -------------------------------------------------------------------------------- /src/cogkit/finetune/diffusion/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/diffusion/trainer.py -------------------------------------------------------------------------------- /src/cogkit/finetune/llm/models/__init__.py: -------------------------------------------------------------------------------- 1 | from . import * # noqa: F403 2 | -------------------------------------------------------------------------------- /src/cogkit/finetune/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/logger.py -------------------------------------------------------------------------------- /src/cogkit/finetune/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/samplers/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/samplers/packing_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/samplers/packing_sampler.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/__init__.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/attention.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/ckpt.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/dist.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/filters.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/io.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/memory.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/misc.py -------------------------------------------------------------------------------- /src/cogkit/finetune/utils/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/finetune/utils/tracker.py -------------------------------------------------------------------------------- /src/cogkit/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/logging.py -------------------------------------------------------------------------------- /src/cogkit/python/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/python/__init__.py -------------------------------------------------------------------------------- /src/cogkit/python/generation/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/python/generation/image.py -------------------------------------------------------------------------------- /src/cogkit/python/generation/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/python/generation/util.py -------------------------------------------------------------------------------- /src/cogkit/python/generation/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/python/generation/video.py -------------------------------------------------------------------------------- /src/cogkit/types/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/types/__init__.py -------------------------------------------------------------------------------- /src/cogkit/types/generation_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/types/generation_mode.py -------------------------------------------------------------------------------- /src/cogkit/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/__init__.py -------------------------------------------------------------------------------- /src/cogkit/utils/diffusion_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/diffusion_pipeline.py -------------------------------------------------------------------------------- /src/cogkit/utils/dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/dtype.py -------------------------------------------------------------------------------- /src/cogkit/utils/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/load.py -------------------------------------------------------------------------------- /src/cogkit/utils/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/lora.py -------------------------------------------------------------------------------- /src/cogkit/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/misc.py -------------------------------------------------------------------------------- /src/cogkit/utils/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/path.py -------------------------------------------------------------------------------- /src/cogkit/utils/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/prompt.py -------------------------------------------------------------------------------- /src/cogkit/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/random.py -------------------------------------------------------------------------------- /src/cogkit/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/src/cogkit/utils/seed.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_expand_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tests/test_expand_list.py -------------------------------------------------------------------------------- /tests/test_mean_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tests/test_mean_filter.py -------------------------------------------------------------------------------- /tests/test_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tests/test_sampler.py -------------------------------------------------------------------------------- /tests/utils/test_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tests/utils/test_path.py -------------------------------------------------------------------------------- /tools/converters/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tools/converters/merge.py -------------------------------------------------------------------------------- /tools/converters/zero2diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tools/converters/zero2diffusers.py -------------------------------------------------------------------------------- /tools/pixelart_dataset/download_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tools/pixelart_dataset/download_convert.py -------------------------------------------------------------------------------- /tools/resize_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/CogKit/HEAD/tools/resize_img.py --------------------------------------------------------------------------------