diff --git a/bridge.py b/bridge.py index c838dd7..30a1e06 100644 --- a/bridge.py +++ b/bridge.py @@ -85,10 +85,10 @@ class LightX2VDefaultConfig: DEFAULT_ATTENTION_TYPE = "flash_attn3" DEFAULT_QUANTIZATION_SCHEMES = { - "dit": "bf16", - "t5": "bf16", - "clip": "fp16", - "adapter": "bf16", + "dit": "Default", + "t5": "Default", + "clip": "Default", + "adapter": "Default", } DEFAULT_VIDEO_PARAMS = { "height": 480, @@ -385,13 +385,13 @@ class ModularConfigManager: updates.update( { - "clip_quantized": clip_scheme != "none", + "clip_quantized": clip_scheme != "Default", "clip_quant_scheme": clip_scheme, - "t5_quantized": t5_scheme != "none", + "t5_quantized": t5_scheme != "Default", "t5_quant_scheme": t5_scheme, - "dit_quantized": dit_scheme != "none", + "dit_quantized": dit_scheme != "Default", "dit_quant_scheme": dit_scheme, - "adapter_quantized": adapter_scheme != "none", + "adapter_quantized": adapter_scheme != "Default", "adapter_quant_scheme": adapter_scheme, } ) diff --git a/data_models.py b/data_models.py index 6bfa767..072b85b 100644 --- a/data_models.py +++ b/data_models.py @@ -95,10 +95,10 @@ class TeaCacheConfig: class QuantizationConfig: """Quantization configuration.""" - dit_quant_scheme: str = "fp8-sgl" - t5_quant_scheme: str = "fp8-sgl" - clip_quant_scheme: str = "fp8-sgl" - adapter_quant_scheme: str = "fp8-sgl" + dit_quant_scheme: str = "Default" + t5_quant_scheme: str = "Default" + clip_quant_scheme: str = "Default" + adapter_quant_scheme: str = "Default" def to_dict(self) -> Dict[str, Any]: return { diff --git a/nodes.py b/nodes.py index de1751c..5bb4061 100644 --- a/nodes.py +++ b/nodes.py @@ -305,7 +305,7 @@ class LightX2VQuantization: quant_backends.append(op_name) common_schema = ["fp8", "int8"] - supported_quant_schemes = ["none"] + supported_quant_schemes = ["Default"] for schema in common_schema: for backend in quant_backends: supported_quant_schemes.append(f"{schema}-{backend}") @@ -329,7 +329,7 @@ class LightX2VQuantization: "clip_quant_scheme": ( supported_quant_schemes, { - "default": supported_quant_schemes[1], + "default": supported_quant_schemes[0], "tooltip": "CLIP encoder quantization precision", }, ),