diff --git a/examples/inference/basic/basic_minimax_h3_fl2va.py b/examples/inference/basic/basic_minimax_h3_fl2va.py index d350cbd5c..f73eabc52 100644 --- a/examples/inference/basic/basic_minimax_h3_fl2va.py +++ b/examples/inference/basic/basic_minimax_h3_fl2va.py @@ -27,7 +27,8 @@ def parse_args() -> argparse.Namespace: parser.add_argument("--model-path", default="MiniMaxAI/MiniMax-H3") # Rank-reduced AdaLN checkpoint (-39% params, -23 GiB VRAM): pass # --model-path noctuashap/MiniMax-H3-pruned-r16 - # (or a local dir converted with tools/minimax_h3/fit_adaln_basis.py). + # (or a local dir produced by + # scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py). # adaln_rank is read from the checkpoint config; no other flags needed. # Rank-reduced checkpoints are inference-only: training needs the # full-rank release. diff --git a/examples/inference/basic/basic_minimax_h3_ref2va.py b/examples/inference/basic/basic_minimax_h3_ref2va.py index 3c485a88c..9cf06996a 100644 --- a/examples/inference/basic/basic_minimax_h3_ref2va.py +++ b/examples/inference/basic/basic_minimax_h3_ref2va.py @@ -27,7 +27,8 @@ def parse_args() -> argparse.Namespace: parser.add_argument("--model-path", default="MiniMaxAI/MiniMax-H3") # Rank-reduced AdaLN checkpoint (-39% params, -23 GiB VRAM): pass # --model-path noctuashap/MiniMax-H3-pruned-r16 - # (or a local dir converted with tools/minimax_h3/fit_adaln_basis.py). + # (or a local dir produced by + # scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py). # adaln_rank is read from the checkpoint config; no other flags needed. # Rank-reduced checkpoints are inference-only: training needs the # full-rank release. diff --git a/examples/inference/basic/basic_minimax_h3_t2v.py b/examples/inference/basic/basic_minimax_h3_t2v.py index 871ecdc65..4af566a3d 100644 --- a/examples/inference/basic/basic_minimax_h3_t2v.py +++ b/examples/inference/basic/basic_minimax_h3_t2v.py @@ -24,7 +24,8 @@ def parse_args() -> argparse.Namespace: parser.add_argument("--model-path", default="MiniMaxAI/MiniMax-H3") # Rank-reduced AdaLN checkpoint (-39% params, -23 GiB VRAM): pass # --model-path noctuashap/MiniMax-H3-pruned-r16 - # (or a local dir converted with tools/minimax_h3/fit_adaln_basis.py). + # (or a local dir produced by + # scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py). # adaln_rank is read from the checkpoint config; no other flags needed. # Rank-reduced checkpoints are inference-only: training needs the # full-rank release. diff --git a/fastvideo/models/dits/minimax_h3.py b/fastvideo/models/dits/minimax_h3.py index bda4209a6..f05f16013 100644 --- a/fastvideo/models/dits/minimax_h3.py +++ b/fastvideo/models/dits/minimax_h3.py @@ -546,7 +546,7 @@ class MiniMaxH3Transformer3DModel(BaseDiT): "parameter, but factorized AdaLN weights are pinned to FP16 " "(BF16 reconstructs them ~1.7x worse). Fine-tune the full-rank " "checkpoint instead, then re-fit the basis with " - "tools/minimax_h3/fit_adaln_basis.py.") + "scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py.") adaln_dim = self.adaln_rank or arch.time_embed_dim self.adaln_basis = ReplicatedLinear( arch.time_embed_dim, diff --git a/scripts/checkpoint_conversion/AGENTS.md b/scripts/checkpoint_conversion/AGENTS.md index 784e92a61..730ea9937 100644 --- a/scripts/checkpoint_conversion/AGENTS.md +++ b/scripts/checkpoint_conversion/AGENTS.md @@ -14,6 +14,7 @@ checkpoint_conversion/ ├── convert_gamecraft_weights.py # DiT only ├── convert_gen3c_to_fastvideo.py ├── convert_ltx2_weights.py +├── convert_minimax_h3_adaln_rank.py # Rank-reduces AdaLN in place; same family ├── convert_turbodiffusion_to_diffusers.py ├── convert_turbodiffusion_i2v_to_diffusers.py ├── extract_llava_text_encoder.py # Encoder extraction from a multimodal repo diff --git a/tools/minimax_h3/fit_adaln_basis.py b/scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py similarity index 99% rename from tools/minimax_h3/fit_adaln_basis.py rename to scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py index 983c4dff9..4a5cdfb4c 100644 --- a/tools/minimax_h3/fit_adaln_basis.py +++ b/scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py @@ -24,7 +24,7 @@ directly: ``adaln_rank`` in ``config.json`` flows onto the arch config through Usage:: - python tools/minimax_h3/fit_adaln_basis.py \ + python scripts/checkpoint_conversion/convert_minimax_h3_adaln_rank.py \ --src /path/to/MiniMax-H3/transformer \ --dst /path/to/MiniMax-H3-r16/transformer --rank 16