Add token to download

convert.py

@@ -20,12 +20,12 @@ from transformers import CONFIG_MAPPING
 COMMIT_MESSAGE = " This PR adds fp32 and fp16 weights in PyTorch and safetensors format to {}"
 
 
-def convert_single(model_id: str, filename: str, model_type: str, sample_size: int, scheduler_type: str, extract_ema: bool, folder: str, progress):
+def convert_single(model_id: str, token:str, filename: str, model_type: str, sample_size: int, scheduler_type: str, extract_ema: bool, folder: str, progress):
     from_safetensors = filename.endswith(".safetensors")
 
     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)
 
     if model_type == "v1":
         config_url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
@@ -81,7 +81,7 @@ def convert(token: str, model_id: str, filename: str, model_type: str, sample_si
         os.makedirs(folder)
         new_pr = None
         try:
-            folder = convert_single(model_id, filename, model_type, sample_size, scheduler_type, extract_ema, folder, progress)
+            folder = convert_single(model_id, token, filename, model_type, sample_size, scheduler_type, extract_ema, folder, progress)
             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]