Updated with ruff formatting

This commit is contained in:
Millun Atluri 2023-12-28 11:05:19 +11:00
parent e38d0e39b7
commit 53b835945f

View File

@ -217,7 +217,7 @@ class ModelPatcher:
for ti_name, ti in ti_list:
ti_embedding = _get_ti_embedding(text_encoder.get_input_embeddings(), ti)
ti_tokens = []
for i in range(ti_embedding.shape[0]):
embedding = ti_embedding[i]