This commit is contained in:
wangshuai09
2024-02-22 20:17:10 +08:00
parent b7aa425344
commit ba66cf8d69
2 changed files with 2 additions and 1 deletions

View File

@@ -230,7 +230,7 @@ class FrozenCLIPEmbedderWithCustomWordsBase(torch.nn.Module):
for fixes in self.hijack.fixes:
for _position, embedding in fixes:
used_embeddings[embedding.name] = embedding
devices.torch_npu_set_device()
z = self.process_tokens(tokens, multipliers)
zs.append(z)