diff --git a/bridge.py b/bridge.py index 1f589e0..dacc1ee 100644 --- a/bridge.py +++ b/bridge.py @@ -317,6 +317,7 @@ class ModularConfigManager: "use_31_block": "use_31_block", "prev_frame_length": "prev_frame_length", "fixed_area": "fixed_area", + "f2v_process": "f2v_process", } self._update_from_config(updates, config, basic_mappings) diff --git a/config_builder.py b/config_builder.py index 2352102..5b2781b 100644 --- a/config_builder.py +++ b/config_builder.py @@ -166,6 +166,9 @@ class InferenceConfigBuilder: if "prev_frame_length" in optional_params: config.prev_frame_length = optional_params["prev_frame_length"] + if "f2v_process" in optional_params: + config.f2v_process = optional_params["f2v_process"] + class TalkObjectConfigBuilder: """Builder for talk object configurations.""" @@ -338,6 +341,7 @@ class ConfigBuilder: "adapter_quant_scheme": getattr(config, "adapter_quant_scheme", None), "adapter_quantized": getattr(config, "adapter_quantized", False), "feature_caching": getattr(config, "feature_caching", None), + "f2v_process": getattr(config, "f2v_process", False), } config_str = json.dumps(relevant_configs, sort_keys=True) diff --git a/data_models.py b/data_models.py index 2022f8b..128a800 100644 --- a/data_models.py +++ b/data_models.py @@ -65,6 +65,7 @@ class InferenceConfig: negative_prompt: str = "" image_path: Optional[str] = None audio_path: Optional[str] = None + f2v_process: bool = False def to_dict(self) -> Dict[str, Any]: """Convert to dictionary, excluding None values.""" diff --git a/nodes.py b/nodes.py index d233015..f84a766 100644 --- a/nodes.py +++ b/nodes.py @@ -199,6 +199,13 @@ class LightX2VInferenceConfig: "tooltip": "Use lightweight VAE to accelerate decoding", }, ), + "f2v_process": ( + "BOOLEAN", + { + "default": False, + "tooltip": "Use f2v process", + }, + ), }, } @@ -227,6 +234,7 @@ class LightX2VInferenceConfig: segment_length=81, prev_frame_length=5, use_tiny_vae=False, + f2v_process=False, ): """Create basic inference configuration.""" builder = InferenceConfigBuilder()