From 1d8275e982cd1532d961b669c14f5365d2c10041 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 06:23:30 -0800
Subject: [PATCH] Update forge_clip.py

---
 modules_forge/forge_clip.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py
index 06ca4474..7079c4b9 100644
--- a/modules_forge/forge_clip.py
+++ b/modules_forge/forge_clip.py
@@ -45,4 +45,6 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords):
         else:
             z = outputs.hidden_states[self.wrapped.layer_idx]
 
+        z.pooled = outputs.pooler_output
+
         return z