From 654264600545e1de252e983644356a0863be1773 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 7 Mar 2023 22:53:05 -0600 Subject: [PATCH] fix(api): match format in schema for sd-concepts --- api/onnx_web/convert/diffusion/textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py index 9dcfda50..92f10532 100644 --- a/api/onnx_web/convert/diffusion/textual_inversion.py +++ b/api/onnx_web/convert/diffusion/textual_inversion.py @@ -32,7 +32,7 @@ def convert_diffusion_textual_inversion( makedirs(path.join(dest_path, "text_encoder"), exist_ok=True) - if format == "huggingface": + if format == "concept": embeds_file = hf_hub_download(repo_id=inversion, filename="learned_embeds.bin") token_file = hf_hub_download(repo_id=inversion, filename="token_identifier.txt")