Update convert.py

convert.py

@@ -113,13 +113,20 @@ def convert_file(
     loaded = torch.load(pt_filename, map_location="cpu")
     if "state_dict" in loaded:
         loaded = loaded["state_dict"]
-    loaded = _remove_duplicate_names(loaded)
+    to_removes = _remove_duplicate_names(loaded)
+
+    metadata = {"format": "pt"}
+    for kept_name, to_remove_group in to_removes.items():
+        for to_remove in to_remove_group:
+            if to_remove not in metadata:
+                metadata[to_remove] = kept_name
+            del loaded[to_remove]
     # For tensors to be contiguous
     loaded = {k: v.contiguous() for k, v in loaded.items()}
 
     dirname = os.path.dirname(sf_filename)
     os.makedirs(dirname, exist_ok=True)
-    save_file(loaded, sf_filename, metadata={"format": "pt"})
+    save_file(loaded, sf_filename, metadata=metadata)
     reloaded = load_file(sf_filename)
     for k in loaded:
         pt_tensor = loaded[k]