Upload clean_merge_vast.sh with huggingface_hub

clean_merge_vast.sh

@@ -40,6 +40,15 @@ model = PeftModel.from_pretrained(base, "/adapter", adapter_name="zerosec_v3")
 merged = model.merge_and_unload()
 
 print("Saving merged...")
+# Fix tensor names with '$' suffix that convert script chokes on
+bad = [k for k in merged.state_dict() if '$' in k]
+if bad:
+    print(f"  Fixing {len(bad)} tensors with '$' in name:")
+    for k in bad[:5]: print(f"    {k}")
+    sd = merged.state_dict()
+    for k in bad:
+        sd[k.replace('$', '')] = sd.pop(k)
+    merged.load_state_dict(sd)
 merged.save_pretrained("/merged_model")
 tok.save_pretrained("/merged_model")