diff --git a/fastvideo/v1/configs/pipelines/base.py b/fastvideo/v1/configs/pipelines/base.py index 29f3cbbf4..3e07e5c5a 100644 --- a/fastvideo/v1/configs/pipelines/base.py +++ b/fastvideo/v1/configs/pipelines/base.py @@ -52,7 +52,7 @@ class PipelineConfig: # VAE configuration vae_config: VAEConfig = field(default_factory=VAEConfig) - vae_precision: str = "fp16" + vae_precision: str = "fp32" vae_tiling: bool = True vae_sp: bool = True diff --git a/fastvideo/v1/configs/pipelines/wan.py b/fastvideo/v1/configs/pipelines/wan.py index 6036802a1..3eaeacf67 100644 --- a/fastvideo/v1/configs/pipelines/wan.py +++ b/fastvideo/v1/configs/pipelines/wan.py @@ -50,7 +50,7 @@ class WanT2V480PConfig(PipelineConfig): # Precision for each component precision: str = "bf16" - vae_precision: str = "fp16" + vae_precision: str = "fp32" text_encoder_precisions: Tuple[str, ...] = field( default_factory=lambda: ("fp32", )) diff --git a/fastvideo/v1/configs/wan_1.3B_t2v_pipeline.json b/fastvideo/v1/configs/wan_1.3B_t2v_pipeline.json index 13e7470d4..a2d959b8c 100644 --- a/fastvideo/v1/configs/wan_1.3B_t2v_pipeline.json +++ b/fastvideo/v1/configs/wan_1.3B_t2v_pipeline.json @@ -4,7 +4,7 @@ "use_cpu_offload": true, "disable_autocast": false, "precision": "bf16", - "vae_precision": "fp16", + "vae_precision": "fp32", "vae_tiling": false, "vae_sp": false, "vae_config": { diff --git a/fastvideo/v1/configs/wan_14B_i2v_480p_pipeline.json b/fastvideo/v1/configs/wan_14B_i2v_480p_pipeline.json index bf908da0c..1035eefe5 100644 --- a/fastvideo/v1/configs/wan_14B_i2v_480p_pipeline.json +++ b/fastvideo/v1/configs/wan_14B_i2v_480p_pipeline.json @@ -4,7 +4,7 @@ "use_cpu_offload": true, "disable_autocast": false, "precision": "bf16", - "vae_precision": "fp16", + "vae_precision": "fp32", "vae_tiling": false, "vae_sp": false, "vae_config": {