Delete modify_safetensors.py

modify_safetensors.py

@@ -1,32 +0,0 @@
-from safetensors import safe_open
-from safetensors.torch import save_file
-import torch
-
-# ==========================================================
-# 1. Load original safetensors
-# ==========================================================
-input_path = "model.safetensors"
-output_path = "model_fixed.safetensors"
-
-new_state_dict = {}
-
-with safe_open(input_path, framework="pt", device="cpu") as f:
-    for key in f.keys():
-        tensor = f.get_tensor(key)
-        # Add prefix "model." if not already
-        new_key = key if key.startswith("model.") else f"model.{key}"
-        new_state_dict[new_key] = tensor
-
-metadata = {
-    "format": "pt",
-}
-# ==========================================================
-# 2. Save to new safetensors file
-# ==========================================================
-save_file(new_state_dict, output_path, metadata=metadata)
-print(f"✅ Saved updated safetensors to {output_path}")
-print(f"✅ Total tensors: {len(new_state_dict)}")
-
-# Optional: verify one sample
-print("Example tensor key:", list(new_state_dict.keys())[0])
-print("Shape:", new_state_dict[list(new_state_dict.keys())[0]].shape)