From b088b9e807349be7ac1d2a19cb3d2ca2e328a0a0 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 08:40:36 -0800
Subject: [PATCH] Update forge_loader.py

---
 modules_forge/forge_loader.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py
index cf160062..45f85d32 100644
--- a/modules_forge/forge_loader.py
+++ b/modules_forge/forge_loader.py
@@ -243,4 +243,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
     sd_model.vae_patcher_original = forge_object.vae.patcher
     timer.record("get patcher")
 
+    # for token counter
+    sd_model.clip = sd_model.cond_stage_model
+
     return sd_model