Comment out unused norm_final in lumina/z image model. (comfyanonymous#11545)

comfyanonymous · web-flow · commit 8fd07170f1b0 · 2025-12-28T22:07:25.000-05:00
diff --git a/comfy/ldm/lumina/model.py b/comfy/ldm/lumina/model.py
@@ -491,7 +491,8 @@ def __init__(
                 for layer_id in range(n_layers)
             ]
         )
-        self.norm_final = operation_settings.get("operations").RMSNorm(dim, eps=norm_eps, elementwise_affine=True, device=operation_settings.get("device"), dtype=operation_settings.get("dtype"))
+        # This norm final is in the lumina 2.0 code but isn't actually used for anything.
+        # self.norm_final = operation_settings.get("operations").RMSNorm(dim, eps=norm_eps, elementwise_affine=True, device=operation_settings.get("device"), dtype=operation_settings.get("dtype"))
         self.final_layer = FinalLayer(dim, patch_size, self.out_channels, z_image_modulation=z_image_modulation, operation_settings=operation_settings)
 
         if self.pad_tokens_multiple is not None:

Original file line number	Diff line number	Diff line change
`@@ -491,7 +491,8 @@ def __init__(`
`491`	`491`	`for layer_id in range(n_layers)`
`492`	`492`	`]`
`493`	`493`	`)`
`494`		`- self.norm_final = operation_settings.get("operations").RMSNorm(dim, eps=norm_eps, elementwise_affine=True, device=operation_settings.get("device"), dtype=operation_settings.get("dtype"))`
	`494`	`+ # This norm final is in the lumina 2.0 code but isn't actually used for anything.`
	`495`	`+ # self.norm_final = operation_settings.get("operations").RMSNorm(dim, eps=norm_eps, elementwise_affine=True, device=operation_settings.get("device"), dtype=operation_settings.get("dtype"))`
`495`	`496`	`self.final_layer = FinalLayer(dim, patch_size, self.out_channels, z_image_modulation=z_image_modulation, operation_settings=operation_settings)`
`496`	`497`
`497`	`498`	`if self.pad_tokens_multiple is not None:`