Update pyramid_dit_for_video_gen_pipeline.py

2024-10-11 00:59:15 +03:00
parent 5252379284
commit 0ef3005e96
1 changed files with 1 additions and 1 deletions
@@ -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)