From 22e9d544082f22a057c8d71c2b6f9cd4dc132aab Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 2 Mar 2023 22:32:37 -0600 Subject: [PATCH] apply lint --- api/onnx_web/convert/diffusion/textual_inversion.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py index 809e580ea..b3214c113 100644 --- a/api/onnx_web/convert/diffusion/textual_inversion.py +++ b/api/onnx_web/convert/diffusion/textual_inversion.py @@ -86,7 +86,9 @@ def convert_diffusion_textual_inversion( for i in range(embeds.shape[0]): layer_embeds = embeds[i] layer_token = token[i] - logger.info("embedding %s vector for layer %s", layer_embeds.shape, layer_token) + logger.info( + "embedding %s vector for layer %s", layer_embeds.shape, layer_token + ) token_id = tokenizer.convert_tokens_to_ids(layer_token) text_encoder.get_input_embeddings().weight.data[token_id] = layer_embeds else: