mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-05 16:51:21 -08:00
Code Issues Projects Releases Packages Wiki Activity

add option

Browse source
This commit is contained in:
hako-mikan 2024-02-09 23:57:46 +09:00 • committed by GitHub
parent 0bc7867ccd
commit 6b3f7039b6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/sd_hijack_clip.py
View file

@ -279,7 +279,9 @@ class FrozenCLIPEmbedderWithCustomWordsBase(torch.nn.Module):
original_mean = z.mean()
z = z * batch_multipliers.reshape(batch_multipliers.shape + (1,)).expand(z.shape)
new_mean = z.mean()
z = z * (original_mean / new_mean)
if not getattr(opts, "disable_normalize_embeddings", False):
z = z * (original_mean / new_mean)
if pooled is not None:
z.pooled = pooled