From a786237752faaf05a6398961c177b39102bf6422 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 09:03:31 -0800 Subject: [PATCH] Update forge_clip.py --- modules_forge/forge_clip.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index 16810b38..04d0a9c2 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -26,8 +26,7 @@ class CLIP_SD_XL_L(FrozenCLIPEmbedderWithCustomWords): super().__init__(wrapped, hijack) def encode_with_transformers(self, tokens): - outputs = self.wrapped.transformer(input_ids=tokens, attention_mask=None, - output_hidden_states=self.wrapped.layer == "hidden") + outputs = self.wrapped.transformer(tokens, output_hidden_states=self.wrapped.layer == "hidden") if self.wrapped.layer == "last": z = outputs.last_hidden_state @@ -46,8 +45,7 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords): self.wrapped.layer_idx = -2 def encode_with_transformers(self, tokens): - outputs = self.wrapped.transformer(input_ids=tokens, attention_mask=None, - output_hidden_states=self.wrapped.layer == "hidden") + outputs = self.wrapped.transformer(tokens, output_hidden_states=self.wrapped.layer == "hidden") if self.wrapped.layer == "last": z = outputs.last_hidden_state