From c0019ab45cfb666e109b2bf891f56be684ec8652 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 12 Jun 2024 22:21:44 +0800
Subject: [PATCH] 'update'

---
 mz_prompt_utils.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/mz_prompt_utils.py b/mz_prompt_utils.py
index 0e3ee8e..8782a34 100644
--- a/mz_prompt_utils.py
+++ b/mz_prompt_utils.py
@@ -147,10 +147,14 @@ class Utils:
         return [[cond, {"pooled_output": pooled}]]
 
     def a1111_clip_text_encode(clip, text):
-        from . import ADV_CLIP_emb_encode
-        cond, pooled = ADV_CLIP_emb_encode.advanced_encode(
-            clip, text, "none", "A1111", w_max=1.0, apply_to_pooled=False)
-        return [[cond, {"pooled_output": pooled}]]
+        try:
+            from . import ADV_CLIP_emb_encode
+            cond, pooled = ADV_CLIP_emb_encode.advanced_encode(
+                clip, text, "none", "A1111", w_max=1.0, apply_to_pooled=False)
+            return [[cond, {"pooled_output": pooled}]]
+        except Exception as e:
+            import nodes
+            return nodes.CLIPTextEncode().encode(clip, text)[0]
 
     def cache_get(key):
         return CACHE_POOL.get(key, None)