From d897364c49f6b6c41fd4c64ec51feba9ce8bbceb Mon Sep 17 00:00:00 2001 From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com> Date: Wed, 11 Feb 2026 23:49:14 +0530 Subject: [PATCH] fix: correct clip_txt_pool variable name typo --- modules/stage_c.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/stage_c.py b/modules/stage_c.py index 45f8bab..8a55313 100755 --- a/modules/stage_c.py +++ b/modules/stage_c.py @@ -155,7 +155,7 @@ def gen_r_embedding(self, r, max_positions=10000): def gen_c_embeddings(self, clip_txt, clip_txt_pooled, clip_img): clip_txt = self.clip_txt_mapper(clip_txt) if len(clip_txt_pooled.shape) == 2: - clip_txt_pool = clip_txt_pooled.unsqueeze(1) + clip_txt_pooled = clip_txt_pooled.unsqueeze(1) if len(clip_img.shape) == 2: clip_img = clip_img.unsqueeze(1) clip_txt_pool = self.clip_txt_pooled_mapper(clip_txt_pooled).view(clip_txt_pooled.size(0), clip_txt_pooled.size(1) * self.c_clip_seq, -1)