From 27efc16ec625246a205d20f1081de987b5d143e3 Mon Sep 17 00:00:00 2001
From: TSC <112517630+LucianoCirino@users.noreply.github.com>
Date: Mon, 4 Sep 2023 17:24:46 -0500
Subject: [PATCH] XY Capsule Model/Clip Clone Patch

---
 efficiency_nodes.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/efficiency_nodes.py b/efficiency_nodes.py
index a2f4e55..cc9f752 100644
--- a/efficiency_nodes.py
+++ b/efficiency_nodes.py
@@ -1479,9 +1479,6 @@ class TSC_KSampler:
 
             # Fill Plot Rows (X)
             for X_index, X in enumerate(X_value):
-                # Reset model parameters to their originals
-                model, clip, refiner_model, refiner_clip = \
-                    clone_or_none(original_model, original_clip, original_refiner_model, original_refiner_clip)
 
                 # Define X parameters and generate labels
                 add_noise, seed, steps, start_at_step, end_at_step, return_with_leftover_noise, cfg,\
@@ -1496,6 +1493,8 @@ class TSC_KSampler:
 
                 if X_type != "Nothing" and Y_type == "Nothing":
                     if X_type == "XY_Capsule":
+                        model, clip, refiner_model, refiner_clip = \
+                            clone_or_none(original_model, original_clip, original_refiner_model, original_refiner_clip)
                         model, clip, vae = X.pre_define_model(model, clip, vae)
 
                     # Models & Conditionings
@@ -1516,13 +1515,11 @@ class TSC_KSampler:
                                        refiner_positive, refiner_negative, latent_image, denoise, vae, vae_decode, sampler_type, xy_capsule=xy_capsule)
 
                 elif X_type != "Nothing" and Y_type != "Nothing":
-                    
                     for Y_index, Y in enumerate(Y_value):
-                        # Reset model parameters to their originals
-                        model, clip, refiner_model, refiner_clip = \
-                            clone_or_none(original_model, original_clip, original_refiner_model, original_refiner_clip)
 
                         if Y_type == "XY_Capsule" and X_type == "XY_Capsule":
+                            model, clip, refiner_model, refiner_clip = \
+                                clone_or_none(original_model, original_clip, original_refiner_model, original_refiner_clip)
                             Y.set_x_capsule(X)
 
                         # Define Y parameters and generate labels