apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): improve logging for inversion tokens

This commit is contained in:
Sean Sube 2023-03-18 12:04:39 -05:00
parent 2cb0a6be3c
commit 5b308d5fc1
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion/textual_inversion.py
View File

@ -95,7 +95,7 @@ def blend_textual_inversions(
# add the tokens to the tokenizer
logger.debug(
"found embeddings for %s tokens: %s", len(embeds.keys()), embeds.keys()
"found embeddings for %s tokens: %s", len(embeds.keys()), list(embeds.keys())
)
num_added_tokens = tokenizer.add_tokens(list(embeds.keys()))
if num_added_tokens == 0: