diff --git a/apps/dreamverse/dreamverse/minimax_h3_generation.py b/apps/dreamverse/dreamverse/minimax_h3_generation.py index 427600d4c..72c6c3926 100644 --- a/apps/dreamverse/dreamverse/minimax_h3_generation.py +++ b/apps/dreamverse/dreamverse/minimax_h3_generation.py @@ -84,7 +84,7 @@ class MiniMaxH3GenerationBackend: ComponentConfig, EngineConfig, GeneratorConfig, - MiniMaxH3Config, + MiniMaxH3Options, OffloadConfig, ParallelismConfig, PipelineSelection, @@ -96,7 +96,7 @@ class MiniMaxH3GenerationBackend: model_path=model_path, pipeline=PipelineSelection( components=ComponentConfig(lora_path=adapter_path, lora_strength=1.0), - minimax_h3=MiniMaxH3Config(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), + minimax_h3=MiniMaxH3Options(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), ), engine=EngineConfig( num_gpus=DREAMVERSE_SP_SIZE, diff --git a/examples/inference/basic/basic_fasth3.py b/examples/inference/basic/basic_fasth3.py index eb8bf81ba..c9b7642dd 100644 --- a/examples/inference/basic/basic_fasth3.py +++ b/examples/inference/basic/basic_fasth3.py @@ -31,7 +31,7 @@ from fastvideo.api import ( EngineConfig, GenerationRequest, GeneratorConfig, - MiniMaxH3Config, + MiniMaxH3Options, OffloadConfig, OutputConfig, ParallelismConfig, @@ -263,7 +263,7 @@ def build_generator_config(args: argparse.Namespace) -> GeneratorConfig: lora_path=getattr(args, "lora_path", None), lora_strength=float(getattr(args, "lora_strength", 1.0)), ), - minimax_h3=MiniMaxH3Config( + minimax_h3=MiniMaxH3Options( sequential_load=args.h3_sequential_load, video_decode_backend=None if args.video_decode_backend == "h3-vae" else args.video_decode_backend, taeh3_checkpoint=args.taeh3_checkpoint, diff --git a/examples/inference/basic/basic_fasth3_simplified.py b/examples/inference/basic/basic_fasth3_simplified.py index 0bd565a39..ea6a3acbd 100644 --- a/examples/inference/basic/basic_fasth3_simplified.py +++ b/examples/inference/basic/basic_fasth3_simplified.py @@ -23,7 +23,7 @@ from fastvideo.api import ( # noqa: E402 EngineConfig, GenerationRequest, GeneratorConfig, - MiniMaxH3Config, + MiniMaxH3Options, OffloadConfig, OutputConfig, ParallelismConfig, @@ -73,7 +73,7 @@ def main() -> None: model_path="MiniMaxAI/MiniMax-H3", pipeline=PipelineSelection( components=ComponentConfig(lora_path=adapter_path, lora_strength=1.0), - minimax_h3=MiniMaxH3Config(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), + minimax_h3=MiniMaxH3Options(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), ), engine=EngineConfig( num_gpus=4, diff --git a/examples/inference/basic/basic_fasth3_simplified_profile.py b/examples/inference/basic/basic_fasth3_simplified_profile.py index a700ed32e..c288498d3 100644 --- a/examples/inference/basic/basic_fasth3_simplified_profile.py +++ b/examples/inference/basic/basic_fasth3_simplified_profile.py @@ -17,7 +17,7 @@ from fastvideo.api import ( EngineConfig, GenerationRequest, GeneratorConfig, - MiniMaxH3Config, + MiniMaxH3Options, OffloadConfig, OutputConfig, ParallelismConfig, @@ -63,7 +63,7 @@ def main() -> None: model_path="MiniMaxAI/MiniMax-H3", pipeline=PipelineSelection( components=ComponentConfig(lora_path=adapter_path, lora_strength=1.0), - minimax_h3=MiniMaxH3Config(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), + minimax_h3=MiniMaxH3Options(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), ), engine=EngineConfig( num_gpus=4, diff --git a/examples/inference/lora/minimax_h3_lora_inference.py b/examples/inference/lora/minimax_h3_lora_inference.py index 9a7c4d913..1762e50fe 100644 --- a/examples/inference/lora/minimax_h3_lora_inference.py +++ b/examples/inference/lora/minimax_h3_lora_inference.py @@ -158,7 +158,7 @@ def main(argv: Sequence[str] | None = None) -> None: from fastvideo.models.loader.lora_patch import DenseLoRAPatch from fastvideo import VideoGenerator from fastvideo.api import (AttentionConfig, CompileConfig, ComponentConfig, EngineConfig, GenerationRequest, - GeneratorConfig, MiniMaxH3Config, OffloadConfig, OutputConfig, ParallelismConfig, + GeneratorConfig, MiniMaxH3Options, OffloadConfig, OutputConfig, ParallelismConfig, PipelineSelection, SamplingConfig) # An adapter carrying to_gate_compress needs the VSA backend, because that is the @@ -182,7 +182,7 @@ def main(argv: Sequence[str] | None = None) -> None: model_path=args.model_path, pipeline=PipelineSelection( components=ComponentConfig(lora_path=args.lora_path, lora_strength=args.lora_strength), - minimax_h3=MiniMaxH3Config(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), + minimax_h3=MiniMaxH3Options(vae_parallel_decode=True, vae_parallel_decode_strategy="gather"), ), engine=EngineConfig( num_gpus=args.num_gpus, diff --git a/fastvideo/api/__init__.py b/fastvideo/api/__init__.py index c2de8bcc8..8a4620ac1 100644 --- a/fastvideo/api/__init__.py +++ b/fastvideo/api/__init__.py @@ -10,10 +10,10 @@ from fastvideo.api.schema import ( GeneratorConfig, GpuPoolConfig, InputConfig, - LTX2Config, - LTX2RefineConfig, - LongCatConfig, - MiniMaxH3Config, + LTX2Options, + LTX2RefineOptions, + LongCatOptions, + MiniMaxH3Options, OffloadConfig, OutputConfig, ParallelismConfig, @@ -75,10 +75,10 @@ __all__ = [ "GeneratorConfig", "GpuPoolConfig", "InputConfig", - "LTX2Config", - "LTX2RefineConfig", - "LongCatConfig", - "MiniMaxH3Config", + "LTX2Options", + "LTX2RefineOptions", + "LongCatOptions", + "MiniMaxH3Options", "OffloadConfig", "OutputConfig", "ParallelismConfig", diff --git a/fastvideo/api/schema.py b/fastvideo/api/schema.py index 164204b51..4735f96ad 100644 --- a/fastvideo/api/schema.py +++ b/fastvideo/api/schema.py @@ -154,7 +154,7 @@ class ComponentConfig: @dataclass -class LTX2RefineConfig: +class LTX2RefineOptions: """Stage-2 refine assets that ``preset_overrides.refine`` and ``components`` do not cover.""" transformer_path: str | None = flat_field("ltx2_refine_transformer_path", None) @@ -163,7 +163,7 @@ class LTX2RefineConfig: @dataclass -class LTX2Config: +class LTX2Options: """LTX-2 settings. ``None`` keeps the model's default.""" vae_spatial_tile_size_in_pixels: int | None = flat_field("ltx2_vae_spatial_tile_size_in_pixels", None) @@ -178,11 +178,11 @@ class LTX2Config: """Draw latent noise in the legacy native order, which earlier SSIM references use.""" use_distilled_sigmas: bool | None = flat_field("ltx2_use_distilled_sigmas", None) """Use the distilled sigma schedule when the checkpoint provides one.""" - refine: LTX2RefineConfig = field(default_factory=LTX2RefineConfig) + refine: LTX2RefineOptions = field(default_factory=LTX2RefineOptions) @dataclass -class MiniMaxH3Config: +class MiniMaxH3Options: """MiniMax-H3 settings. ``None`` keeps the model's default.""" sequential_load: bool | None = flat_field("h3_sequential_load", None) @@ -207,7 +207,7 @@ class MiniMaxH3Config: @dataclass -class LongCatConfig: +class LongCatOptions: """LongCat block sparse attention (BSA) settings. ``None`` keeps the model's default.""" enable_bsa: bool | None = flat_field("enable_bsa", None) @@ -239,9 +239,9 @@ class PipelineSelection: """Overrides for fields of the model's DiT config, such as ``prefix``.""" vae: dict[str, Any] = field(default_factory=dict) """Overrides for fields of the model's VAE config, such as ``load_encoder`` or ``use_tiling``.""" - ltx2: LTX2Config = field(default_factory=LTX2Config) - minimax_h3: MiniMaxH3Config = field(default_factory=MiniMaxH3Config) - longcat: LongCatConfig = field(default_factory=LongCatConfig) + ltx2: LTX2Options = field(default_factory=LTX2Options) + minimax_h3: MiniMaxH3Options = field(default_factory=MiniMaxH3Options) + longcat: LongCatOptions = field(default_factory=LongCatOptions) preset_overrides: dict[str, Any] = field(default_factory=dict) experimental: dict[str, Any] = field(default_factory=dict) @@ -436,10 +436,10 @@ __all__ = [ "GeneratorConfig", "GpuPoolConfig", "InputConfig", - "LTX2Config", - "LTX2RefineConfig", - "LongCatConfig", - "MiniMaxH3Config", + "LTX2Options", + "LTX2RefineOptions", + "LongCatOptions", + "MiniMaxH3Options", "OffloadConfig", "OutputConfig", "ParallelismConfig",