Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ruff github format errors

Browse Source
This commit is contained in:
Lincoln Stein 2023-12-01 01:59:26 -05:00
parent f95ce1870c
commit 5a3f1f2b22
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_management/lora.py
View File

@ -215,7 +215,7 @@ class ModelPatcher:
text_encoder.resize_token_embeddings(init_tokens_count + new_tokens_added, pad_to_multiple_of) text_encoder.resize_token_embeddings(init_tokens_count + new_tokens_added, pad_to_multiple_of)
model_embeddings = text_encoder.get_input_embeddings() model_embeddings = text_encoder.get_input_embeddings()
for ti_name, ti in ti_list: for ti_name, _ in ti_list:
ti_tokens = [] ti_tokens = []
for i in range(ti_embedding.shape[0]): for i in range(ti_embedding.shape[0]):
embedding = ti_embedding[i] embedding = ti_embedding[i]
@ -537,7 +537,7 @@ class ONNXModelPatcher:
axis=0, axis=0,
) )
for ti_name, ti in ti_list: for ti_name, _ in ti_list:
ti_tokens = [] ti_tokens = []
for i in range(ti_embedding.shape[0]): for i in range(ti_embedding.shape[0]):
embedding = ti_embedding[i].detach().numpy() embedding = ti_embedding[i].detach().numpy()