Upload clean_merge_vast.sh with huggingface_hub

clean_merge_vast.sh

@@ -40,18 +40,25 @@ 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")
 
+# Check which tensors have '$' suffix
+bad = [k for k in merged.state_dict() if '$' in k[-3:]]
+if bad:
+    print(f"  WARNING: {len(bad)} tensors with '\$' suffix — patching converter for these")
+
+print("[3] Patching llama.cpp converter for '\$' tensor names...")
+import re as _re
+base_py = "/llama.cpp/conversion/base.py"
+txt = open(base_py).read()
+txt = txt.replace(
+    "new_name = self.map_tensor_name(name)",
+    "name_clean = name.rstrip('\$'); new_name = self.map_tensor_name(name_clean)"
+)
+open(base_py, "w").write(txt)
+print("  Patched")
+
 # Verify tensor count
 from safetensors import safe_open
 total = 0