Spaces:
Build error
Build error
missing token: str
Browse files- 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 |
|