diff --git a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
index fae7962..93fde78 100644
--- a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
+++ b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
@@ -75,7 +75,7 @@ class PyramidDiTForVideoGeneration:
                     if name != "pos_embedding":
                         param.data = param.data.to(model_dtype)
 
-        if fp8_fastmode == "fastmode":
+        if fp8_fastmode:
             from ..fp8_optimization import convert_fp8_linear
             convert_fp8_linear(self.dit, torch.bfloat16)