lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 17:29:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 09:36:41 -08:00
parent a786237752
commit 836b2164d8
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/forge_clip.py
View File

@@ -1,3 +1,5 @@
import torch
from modules.sd_hijack_clip import FrozenCLIPEmbedderWithCustomWords
from ldm_patched.modules import model_management
@@ -52,6 +54,8 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords):
else:
z = outputs.hidden_states[self.wrapped.layer_idx]
z.pooled = outputs.pooler_output
pooled_output = outputs.pooler_output
text_projection = self.wrapped.text_projection
pooled_output = pooled_output.float().to(text_projection.device) @ text_projection.float()
z.pooled = pooled_output
return z