From 0ef3005e96741a5d0367becd9b66386d585f4691 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?= <40791699+kijai@users.noreply.github.com> Date: Fri, 11 Oct 2024 00:59:15 +0300 Subject: [PATCH] Update pyramid_dit_for_video_gen_pipeline.py --- pyramid_dit/pyramid_dit_for_video_gen_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)