├── .gitignore ├── LICENSE ├── README.md ├── assets ├── demo1.gif ├── icon.png └── methods_overview.gif ├── demos ├── api_example.py ├── cli_test_ti2v_release.py ├── cli_test_ti2v_release.sh └── test_encoder_decoder.py ├── pyproject.toml ├── pyrightconfig.json ├── requirements.txt ├── scripts ├── download_weights.py ├── format.bash ├── pytorch_to_safe_tensors.py ├── typecheck.bash └── weights_to_fp8.py ├── src └── genmo │ ├── lib │ ├── __pycache__ │ │ ├── attn_imports.cpython-312.pyc │ │ ├── progress.cpython-312.pyc │ │ └── utils.cpython-312.pyc │ ├── attn_imports.py │ ├── progress.py │ └── utils.py │ └── mochi_preview │ ├── __init__.py │ ├── __pycache__ │ └── __init__.cpython-312.pyc │ ├── convert_checkpoint.sh │ ├── datasets │ ├── __pycache__ │ │ └── embedding_datasets.cpython-312.pyc │ ├── embedding_datasets.py │ └── video_transforms.py │ ├── dit │ └── joint_model │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-312.pyc │ │ ├── asymm_models_joint.cpython-312.pyc │ │ ├── context_parallel.cpython-312.pyc │ │ ├── layers.cpython-312.pyc │ │ ├── mod_rmsnorm.cpython-312.pyc │ │ ├── residual_tanh_gated_rmsnorm.cpython-312.pyc │ │ ├── rope_mixed.cpython-312.pyc │ │ ├── temporal_rope.cpython-312.pyc │ │ └── utils.cpython-312.pyc │ │ ├── asymm_models_joint.py │ │ ├── context_parallel.py │ │ ├── layers.py │ │ ├── mod_rmsnorm.py │ │ ├── residual_tanh_gated_rmsnorm.py │ │ ├── rope_mixed.py │ │ ├── temporal_rope.py │ │ └── utils.py │ ├── dit_pusa │ └── joint_model │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-312.pyc │ │ ├── asymm_models_joint.cpython-312.pyc │ │ ├── layers.cpython-312.pyc │ │ ├── mod_rmsnorm.cpython-312.pyc │ │ ├── residual_tanh_gated_rmsnorm.cpython-312.pyc │ │ ├── rope_mixed.cpython-312.pyc │ │ ├── temporal_rope.cpython-312.pyc │ │ └── utils.cpython-312.pyc │ │ ├── asymm_models_joint.py │ │ ├── context_parallel.py │ │ ├── layers.py │ │ ├── mod_rmsnorm.py │ │ ├── residual_tanh_gated_rmsnorm.py │ │ ├── rope_mixed.py │ │ ├── temporal_rope.py │ │ └── utils.py │ ├── gradio_ui.py │ ├── pipelines.py │ ├── pipelines_original.py │ ├── pipelines_ti2v_release.py │ ├── save_safetytensors.py │ ├── sharing │ ├── train_mochi_pl_sharing_20250108.py │ └── train_mochi_slurm_sharing.sh │ ├── train_mochi.py │ ├── train_multi_nodes.sh │ ├── train_pusa.py │ └── vae │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-311.pyc │ ├── __init__.cpython-312.pyc │ ├── cp_conv.cpython-310.pyc │ ├── cp_conv.cpython-311.pyc │ ├── cp_conv.cpython-312.pyc │ ├── latent_dist.cpython-310.pyc │ ├── latent_dist.cpython-311.pyc │ ├── latent_dist.cpython-312.pyc │ ├── models.cpython-310.pyc │ ├── models.cpython-311.pyc │ ├── models.cpython-312.pyc │ ├── vae_stats.cpython-310.pyc │ ├── vae_stats.cpython-311.pyc │ └── vae_stats.cpython-312.pyc │ ├── cp_conv.py │ ├── cp_conv_20250218.py │ ├── latent_dist.py │ ├── models.py │ ├── models_20250218.py │ └── vae_stats.py └── uv.lock /.gitignore: -------------------------------------------------------------------------------- 1 | /pusa/ 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/README.md -------------------------------------------------------------------------------- /assets/demo1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/assets/demo1.gif -------------------------------------------------------------------------------- /assets/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/assets/icon.png -------------------------------------------------------------------------------- /assets/methods_overview.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/assets/methods_overview.gif -------------------------------------------------------------------------------- /demos/api_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/demos/api_example.py -------------------------------------------------------------------------------- /demos/cli_test_ti2v_release.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/demos/cli_test_ti2v_release.py -------------------------------------------------------------------------------- /demos/cli_test_ti2v_release.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/demos/cli_test_ti2v_release.sh -------------------------------------------------------------------------------- /demos/test_encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/demos/test_encoder_decoder.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pyrightconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/pyrightconfig.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/download_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/scripts/download_weights.py -------------------------------------------------------------------------------- /scripts/format.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/scripts/format.bash -------------------------------------------------------------------------------- /scripts/pytorch_to_safe_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/scripts/pytorch_to_safe_tensors.py -------------------------------------------------------------------------------- /scripts/typecheck.bash: -------------------------------------------------------------------------------- 1 | #! /bin/bash 2 | npx pyright -------------------------------------------------------------------------------- /scripts/weights_to_fp8.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/genmo/lib/__pycache__/attn_imports.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/__pycache__/attn_imports.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/lib/__pycache__/progress.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/__pycache__/progress.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/lib/__pycache__/utils.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/__pycache__/utils.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/lib/attn_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/attn_imports.py -------------------------------------------------------------------------------- /src/genmo/lib/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/progress.py -------------------------------------------------------------------------------- /src/genmo/lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/lib/utils.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/genmo/mochi_preview/__pycache__/__init__.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/__pycache__/__init__.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/convert_checkpoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/convert_checkpoint.sh -------------------------------------------------------------------------------- /src/genmo/mochi_preview/datasets/__pycache__/embedding_datasets.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/datasets/__pycache__/embedding_datasets.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/datasets/embedding_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/datasets/embedding_datasets.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/datasets/video_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/datasets/video_transforms.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/__init__.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/__init__.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/asymm_models_joint.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/asymm_models_joint.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/context_parallel.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/context_parallel.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/layers.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/layers.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/mod_rmsnorm.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/mod_rmsnorm.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/residual_tanh_gated_rmsnorm.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/residual_tanh_gated_rmsnorm.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/rope_mixed.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/rope_mixed.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/temporal_rope.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/temporal_rope.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/__pycache__/utils.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/__pycache__/utils.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/asymm_models_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/asymm_models_joint.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/context_parallel.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/layers.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/mod_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/mod_rmsnorm.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/residual_tanh_gated_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/residual_tanh_gated_rmsnorm.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/rope_mixed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/rope_mixed.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/temporal_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/temporal_rope.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit/joint_model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit/joint_model/utils.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/__init__.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/__init__.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/asymm_models_joint.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/asymm_models_joint.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/layers.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/layers.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/mod_rmsnorm.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/mod_rmsnorm.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/residual_tanh_gated_rmsnorm.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/residual_tanh_gated_rmsnorm.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/rope_mixed.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/rope_mixed.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/temporal_rope.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/temporal_rope.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/utils.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/__pycache__/utils.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/asymm_models_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/asymm_models_joint.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/context_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/context_parallel.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/layers.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/mod_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/mod_rmsnorm.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/residual_tanh_gated_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/residual_tanh_gated_rmsnorm.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/rope_mixed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/rope_mixed.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/temporal_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/temporal_rope.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/dit_pusa/joint_model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/dit_pusa/joint_model/utils.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/gradio_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/gradio_ui.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/pipelines.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/pipelines_original.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/pipelines_original.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/pipelines_ti2v_release.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/pipelines_ti2v_release.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/save_safetytensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/save_safetytensors.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/sharing/train_mochi_pl_sharing_20250108.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/sharing/train_mochi_pl_sharing_20250108.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/sharing/train_mochi_slurm_sharing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/sharing/train_mochi_slurm_sharing.sh -------------------------------------------------------------------------------- /src/genmo/mochi_preview/train_mochi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/train_mochi.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/train_multi_nodes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/train_multi_nodes.sh -------------------------------------------------------------------------------- /src/genmo/mochi_preview/train_pusa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/train_pusa.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-311.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/__init__.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-310.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-311.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/cp_conv.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-310.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-311.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/latent_dist.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/models.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/models.cpython-310.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/models.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/models.cpython-311.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/models.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/models.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-310.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-311.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-312.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/__pycache__/vae_stats.cpython-312.pyc -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/cp_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/cp_conv.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/cp_conv_20250218.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/cp_conv_20250218.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/latent_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/latent_dist.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/models.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/models_20250218.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/models_20250218.py -------------------------------------------------------------------------------- /src/genmo/mochi_preview/vae/vae_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/src/genmo/mochi_preview/vae/vae_stats.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yaofang-Liu/Mochi-Full-Finetuner/HEAD/uv.lock --------------------------------------------------------------------------------