lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-23 15:59:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_clip.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 10:46:35 -08:00
parent d2f5e8c353
commit f572050e41
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
modules_forge/forge_clip.py
View File

@@ -22,7 +22,15 @@ class CLIP_SD_21_H(FrozenCLIPEmbedderWithCustomWords):
def encode_with_transformers(self, tokens):
model_management.load_models_gpu([self.patcher.patcher])
return super().encode_with_transformers(tokens)
outputs = self.wrapped.transformer(tokens, output_hidden_states=self.wrapped.layer == "hidden")
if self.wrapped.layer == "last":
z = outputs.last_hidden_state
else:
z = outputs.hidden_states[self.wrapped.layer_idx]
z = self.wrapped.transformer.text_model.final_layer_norm(z)
return z
class CLIP_SD_XL_L(FrozenCLIPEmbedderWithCustomWords):