lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-07 00:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_hijack_clip.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 06:10:11 -08:00
parent cc2a88355a
commit fdef0aeb6b
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
modules/sd_hijack_clip.py
View File

@@ -345,6 +345,14 @@ class FrozenCLIPEmbedderWithCustomWords(FrozenCLIPEmbedderWithCustomWordsBase):
return embedded
class CLIP_SD_15_L(FrozenCLIPEmbedderWithCustomWords):
pass
class CLIP_SD_21_G(FrozenCLIPEmbedderWithCustomWords):
pass
class CLIP_SD_XL_L(FrozenCLIPEmbedderWithCustomWords):
def __init__(self, wrapped, hijack):
super().__init__(wrapped, hijack)
@@ -358,3 +366,18 @@ class CLIP_SD_XL_L(FrozenCLIPEmbedderWithCustomWords):
z = outputs.hidden_states[self.wrapped.layer_idx]
return z
class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords):
def __init__(self, wrapped, hijack):
super().__init__(wrapped, hijack)
def encode_with_transformers(self, tokens):
outputs = self.wrapped.transformer(input_ids=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]
return z