appending .value to learnable registers

Author: prishajain1

src/maxdiffusion/models/ltx2/ltx2_utils.py

@@ -415,6 +415,10 @@ def rename_for_ltx2_connector(key):
             key = key.replace(".weight", ".scale")
         else:
             key = key.replace(".weight", ".kernel")
+            
+    if "learnable_registers" in key and not key.endswith(".value"):
+        key = key + ".value"
+        
     return key
 
 def load_connector_weights(