suggestions and fixes from the PR

This commit is contained in:
AUTOMATIC
2023-05-10 21:21:32 +03:00
parent d25219b7e8
commit 3ec7b705c7
11 changed files with 16 additions and 31 deletions

View File

@@ -223,7 +223,7 @@ class FrozenCLIPEmbedderWithCustomWordsBase(torch.nn.Module):
self.hijack.fixes = [x.fixes for x in batch_chunk]
for fixes in self.hijack.fixes:
for position, embedding in fixes: # noqa: B007
for _position, embedding in fixes:
used_embeddings[embedding.name] = embedding
z = self.process_tokens(tokens, multipliers)