idlebg commited on
Commit
7055030
1 Parent(s): 5440daa

missing token: str

Browse files
Files changed (1) hide show
  1. convert.py +2 -2
convert.py CHANGED
@@ -19,10 +19,10 @@ from transformers import CONFIG_MAPPING
19
  COMMIT_MESSAGE = " This PR adds fp32 and fp16 weights in safetensors format to {}"
20
 
21
 
22
- def convert_single(model_id: str, filename: str, folder: str, progress: Any):
23
  progress(0, desc="Downloading model")
24
  local_file = os.path.join(model_id, filename)
25
- ckpt_file = local_file if os.path.isfile(local_file) else hf_hub_download(repo_id=model_id, filename=filename)
26
 
27
  pipeline = StableDiffusionXLPipeline.from_single_file(ckpt_file)
28
 
 
19
  COMMIT_MESSAGE = " This PR adds fp32 and fp16 weights in safetensors format to {}"
20
 
21
 
22
+ def convert_single(model_id: str, filename: str, folder: str, progress: Any, token: str):
23
  progress(0, desc="Downloading model")
24
  local_file = os.path.join(model_id, filename)
25
+ ckpt_file = local_file if os.path.isfile(local_file) else hf_hub_download(repo_id=model_id, filename=filename, token=token)
26
 
27
  pipeline = StableDiffusionXLPipeline.from_single_file(ckpt_file)
28