fix(api): match format in schema for sd-concepts

ssube · Mar 8, 2023 · 6542646 · 6542646
1 parent 85ff700
commit 6542646
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")