More robust with tied weight keys

convert.py

@@ -241,7 +241,7 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str, revision=Optional[st
 
 
 def convert_generic(
-    model_id: str, *, revision=Optional[str], folder: str, filenames: Set[str], token: Optional[str]
+    model_id: str, *, revision=Optional[str], folder: str, filenames: Set[str], token: Optional[str], discard_names: List[str],
 ) -> ConversionResult:
     operations = []
     errors = []
@@ -262,7 +262,7 @@ def convert_generic(
                 sf_in_repo = f"{prefix}.safetensors"
             sf_filename = os.path.join(folder, sf_in_repo)
             try:
-                convert_file(pt_filename, sf_filename, discard_names=[])
+                convert_file(pt_filename, sf_filename, discard_names=discard_names)
                 operations.append(CommitOperationAdd(path_in_repo=sf_in_repo, path_or_fileobj=sf_filename))
             except Exception as e:
                 errors.append((pt_filename, e))
@@ -275,6 +275,10 @@ def convert(
     pr_title = "Adding `safetensors` variant of this model"
     info = api.model_info(model_id, revision=revision)
     filenames = set(s.rfilename for s in info.siblings)
+    try:
+        discard_names = get_discard_names(model_id, revision=revision, folder=folder, token=api.token)
+    except:
+        discard_names = []
 
     with TemporaryDirectory() as d:
         folder = os.path.join(d, repo_folder_name(repo_id=model_id, repo_type="models"))
@@ -283,6 +287,7 @@ def convert(
         try:
             operations = None
             pr = previous_pr(api, model_id, pr_title, revision=revision)
+            
 
             library_name = getattr(info, "library_name", None)
             if any(filename.endswith(".safetensors") for filename in filenames) and not force:
@@ -293,7 +298,6 @@ def convert(
                 raise AlreadyExists(f"Model {model_id} already has an open PR check out {url}")
             elif library_name == "transformers":
 
-                discard_names = get_discard_names(model_id, revision=revision, folder=folder, token=api.token)
                 if "pytorch_model.bin" in filenames:
                     operations, errors = convert_single(
                         model_id, revision=revision, folder=folder, token=api.token, discard_names=discard_names
@@ -306,7 +310,7 @@ def convert(
                     raise RuntimeError(f"Model {model_id} doesn't seem to be a valid pytorch model. Cannot convert")
             else:
                 operations, errors = convert_generic(
-                    model_id, revision=revision, folder=folder, filenames=filenames, token=api.token
+                    model_id, revision=revision, folder=folder, filenames=filenames, token=api.token, discard_names=discard_names
                 )
 
             if operations: