From 1d9fe8c695009925ed729f76d3543f34df8f8ca4 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 7 Nov 2024 21:50:56 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 2826a8c..da230c9 100644
--- a/nodes.py
+++ b/nodes.py
@@ -224,6 +224,7 @@ class PyramidFlowModelLoader:
 
         return (pyramid_model,)
 
+#region Sampler
 class PyramidFlowSampler:
     @classmethod
     def INPUT_TYPES(s):
@@ -358,7 +359,7 @@ class PyramidFlowTextEncode:
 
         if force_offload:
             clip.cond_stage_model.to(offload_device)
-        
+        clip.cond_stage_model.reset_clip_options()
         embeds = {
             "prompt_embeds": prompt_embeds.to(device),
             "attention_mask": prompt_attention_mask["attention_mask"].to(device),
@@ -369,6 +370,8 @@ class PyramidFlowTextEncode:
         }
 
         return (embeds, )
+    
+#region VAE
 class PyramidFlowVAEEncode:
     @classmethod
     def INPUT_TYPES(s):