Fix issue loading fp8 ltxav checkpoints. (Comfy-Org#12582)

comfyanonymous · web-flow · commit caa43d2395a6 · 2026-02-22T16:00:02.000-05:00
diff --git a/comfy/ldm/lightricks/embeddings_connector.py b/comfy/ldm/lightricks/embeddings_connector.py
@@ -157,11 +157,9 @@ def __init__(
         self.num_learnable_registers = num_learnable_registers
         if self.num_learnable_registers:
             self.learnable_registers = nn.Parameter(
-                torch.rand(
+                torch.empty(
                     self.num_learnable_registers, inner_dim, dtype=dtype, device=device
                 )
-                * 2.0
-                - 1.0
             )
 
     def get_fractional_positions(self, indices_grid):

Original file line number	Diff line number	Diff line change
`@@ -157,11 +157,9 @@ def __init__(`
`157`	`157`	`self.num_learnable_registers = num_learnable_registers`
`158`	`158`	`if self.num_learnable_registers:`
`159`	`159`	`self.learnable_registers = nn.Parameter(`
`160`		`- torch.rand(`
	`160`	`+ torch.empty(`
`161`	`161`	`self.num_learnable_registers, inner_dim, dtype=dtype, device=device`
`162`	`162`	`)`
`163`		`- * 2.0`
`164`		`- - 1.0`
`165`	`163`	`)`
`166`	`164`
`167`	`165`	`def get_fractional_positions(self, indices_grid):`