removed aesthetic gradients as built-in

added support for extensions
This commit is contained in:
AUTOMATIC
2022-10-22 12:23:45 +03:00
parent 26d1073745
commit 2b91251637
14 changed files with 249 additions and 410 deletions

View File

@@ -332,7 +332,6 @@ class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module):
multipliers.append([1.0] * 75)
z1 = self.process_tokens(tokens, multipliers)
z1 = shared.aesthetic_clip(z1, remade_batch_tokens)
z = z1 if z is None else torch.cat((z, z1), axis=-2)
remade_batch_tokens = rem_tokens