Fix issue loading fp8 ltxav checkpoints. (#12582)

2026-02-28 02:44:04 +00:00 · 2026-02-22 13:00:02 -08:00
parent 07ca6852e8
commit caa43d2395
1 changed files with 1 additions and 3 deletions
--- a/comfy/ldm/lightricks/embeddings_connector.py
+++ b/comfy/ldm/lightricks/embeddings_connector.py
@@ -157,11 +157,9 @@ class Embeddings1DConnector(nn.Module):
        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):