From 30b08c6d6d3af2e2171702347649e67a65baf28e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 7 Mar 2023 23:05:53 -0600 Subject: [PATCH] fix(api): log correct token when converting Textual Inversions --- api/onnx_web/convert/diffusion/textual_inversion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py index 92f10532..90ec28bc 100644 --- a/api/onnx_web/convert/diffusion/textual_inversion.py +++ b/api/onnx_web/convert/diffusion/textual_inversion.py @@ -60,7 +60,7 @@ def convert_diffusion_textual_inversion( else: 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( base_model, @@ -92,7 +92,7 @@ def convert_diffusion_textual_inversion( for i in range(embeds.shape[0]): layer_embeds = embeds[i] layer_token = token[i] - logger.info( + logger.debug( "embedding %s vector for layer %s", layer_embeds.shape, layer_token ) token_id = tokenizer.convert_tokens_to_ids(layer_token)