Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

add clip-skip notes

Browse Source
This commit is contained in:
Vladimir Mandic
2024-02-09 14:29:17 -05:00
parent a069a45e6a
commit 40feb79b11
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_hijack_clip.py
+2 -1
View File
@@ -240,7 +240,8 @@ class FrozenCLIPEmbedderWithCustomWords(FrozenCLIPEmbedderWithCustomWordsBase):
return tokenized
def encode_with_transformers(self, tokens):
clip_skip = opts.data['clip_skip'] or 1
clip_skip = int(opts.data['clip_skip']) or 1
print('HERE', type(clip_skip), clip_skip)
outputs = self.wrapped.transformer(input_ids=tokens, output_hidden_states=-clip_skip)
if clip_skip > 1:
z = outputs.hidden_states[-clip_skip]