token: str added for private repos (#1)

- token: str added for private repos (f7566a08e6ee7c660df91bf2f24a81bc628655aa)


Co-authored-by: idle stoev <[email protected]>

convert.py

@@ -19,10 +19,10 @@ from transformers import CONFIG_MAPPING
 COMMIT_MESSAGE = " This PR adds fp32 and fp16 weights in safetensors format to {}"
 
 
-def convert_single(model_id: str, filename: str, folder: str, progress: Any):
+def convert_single(model_id: str, filename: str, folder: str, progress: Any, token: str):
     progress(0, desc="Downloading model")
     local_file = os.path.join(model_id, filename)
-    ckpt_file = local_file if os.path.isfile(local_file) else hf_hub_download(repo_id=model_id, filename=filename)
+    ckpt_file = local_file if os.path.isfile(local_file) else hf_hub_download(repo_id=model_id, filename=filename, token=token)
 
     pipeline = StableDiffusionXLPipeline.from_single_file(ckpt_file)
 
@@ -55,7 +55,7 @@ def convert(token: str, model_id: str, filename: str, progress=gr.Progress()):
         os.makedirs(folder)
         new_pr = None
         try:
-            folder = convert_single(model_id, filename, folder, progress)
+            folder = convert_single(model_id, filename, folder, progress, token)
             progress(0.7, desc="Uploading to Hub")
             new_pr  = api.upload_folder(folder_path=folder, path_in_repo="./", repo_id=model_id, repo_type="model", token=token, commit_message=pr_title, commit_description=COMMIT_MESSAGE.format(model_id), create_pr=True)
             pr_number = new_pr.split("%2F")[-1].split("/")[0]
@@ -66,4 +66,4 @@ def convert(token: str, model_id: str, filename: str, progress=gr.Progress()):
         finally:
             shutil.rmtree(folder)
 
-        return link
+        return link