Update convert.py

convert.py

@@ -45,7 +45,7 @@ def rename(pt_filename: str) -> str:
     return local
 
 
-def convert_multi(model_id: str, folder: str) -> ConversionResult:
+def convert_multi(model_id: str, folder: str, api: "HfApi") -> ConversionResult:
     filename = hf_hub_download(repo_id=model_id, filename="pytorch_model.bin.index.json")
     with open(filename, "r") as f:
         data = json.load(f)
@@ -87,7 +87,7 @@ def convert_multi(model_id: str, folder: str) -> ConversionResult:
 
 
 
-def convert_single(model_id: str, folder: str) -> ConversionResult:
+def convert_single(model_id: str, folder: str, api: "HfApi") -> ConversionResult:
     pt_filename = hf_hub_download(repo_id=model_id, filename="pytorch_model.bin")
 
     sf_name = "model.safetensors"
@@ -157,7 +157,7 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str) -> Optional["Discuss
     return None
 
 
-def convert_generic(model_id: str, folder: str, filenames: Set[str]) -> ConversionResult:
+def convert_generic(model_id: str, folder: str, filenames: Set[str], api: "HfApi") -> ConversionResult:
     operations = []
     errors = []
 
@@ -224,13 +224,13 @@ def convert(api: "HfApi", model_id: str, force: bool = False) -> Tuple["CommitIn
                 raise AlreadyExists(f"Model {model_id} already has an open PR check out {url}")
             elif library_name == "transformers":
                 if "pytorch_model.bin" in filenames:
-                    new_pr = convert_single(model_id, folder)
+                    new_pr = convert_single(model_id, folder, api)
                 elif "pytorch_model.bin.index.json" in filenames:
-                    new_pr = convert_multi(model_id, folder)
+                    new_pr = convert_multi(model_id, folder, api)
                 else:
                     raise RuntimeError(f"Model {model_id} doesn't seem to be a valid pytorch model. Cannot convert")
             else:
-                new_pr = convert_generic(model_id, folder, filenames)
+                new_pr = convert_generic(model_id, folder, filenames, api)
 
             print(f"Pr created at {new_pr.pr_url}")
         finally: