Only enable fp16 on z image models that actually support it. (#12065)

2026-01-24 19:32:28 -08:00
parent ed6002cb60
commit 635406e283
3 changed files with 6 additions and 1 deletions
--- a/comfy/ldm/lumina/model.py
+++ b/comfy/ldm/lumina/model.py
@@ -451,6 +451,7 @@ class NextDiT(nn.Module):
        device=None,
        dtype=None,
        operations=None,
+        **kwargs,
    ) -> None:
        super().__init__()
        self.dtype = dtype