apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): log correct token when converting Textual Inversions

This commit is contained in:
Sean Sube 2023-03-07 23:05:53 -06:00
parent 6542646005
commit 30b08c6d6d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -60,7 +60,7 @@ def convert_diffusion_textual_inversion(
else: else:
raise ValueError(f"unknown textual inversion format: {format}") raise ValueError(f"unknown textual inversion format: {format}")
logger.info("found embedding for token %s: %s", trained_token, embeds.shape) logger.info("found embeddings for token %s: %s", token, embeds.shape)
tokenizer = CLIPTokenizer.from_pretrained( tokenizer = CLIPTokenizer.from_pretrained(
base_model, base_model,
@ -92,7 +92,7 @@ def convert_diffusion_textual_inversion(
for i in range(embeds.shape[0]): for i in range(embeds.shape[0]):
layer_embeds = embeds[i] layer_embeds = embeds[i]
layer_token = token[i] layer_token = token[i]
logger.info( logger.debug(
"embedding %s vector for layer %s", layer_embeds.shape, layer_token "embedding %s vector for layer %s", layer_embeds.shape, layer_token
) )
token_id = tokenizer.convert_tokens_to_ids(layer_token) token_id = tokenizer.convert_tokens_to_ids(layer_token)