apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-03-02 22:32:37 -06:00
parent be1c8cd79f
commit 22e9d54408
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 1 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

@ -86,7 +86,9 @@ 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("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) token_id = tokenizer.convert_tokens_to_ids(layer_token)
text_encoder.get_input_embeddings().weight.data[token_id] = layer_embeds text_encoder.get_input_embeddings().weight.data[token_id] = layer_embeds
else: else: