From 7f2005127ff20170e2d92f353416c7f0705c593b Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 11 Mar 2023 14:52:29 +0300
Subject: [PATCH] rename CFGDenoiserParams fields for #8064

---
 modules/script_callbacks.py       | 10 +++++-----
 modules/sd_samplers_kdiffusion.py |  4 ++--
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index d17031355..079118761 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -29,7 +29,7 @@ class ImageSaveParams:
 
 
 class CFGDenoiserParams:
-    def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, tensor, uncond):
+    def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, text_cond, text_uncond):
         self.x = x
         """Latent image representation in the process of being denoised"""
         
@@ -45,11 +45,11 @@ class CFGDenoiserParams:
         self.total_sampling_steps = total_sampling_steps
         """Total number of sampling steps planned"""
         
-        self.tensor = tensor
-        """ Encoder hidden states of conditioning"""
+        self.text_cond = text_cond
+        """ Encoder hidden states of text conditioning from prompt"""
         
-        self.uncond = uncond
-        """ Encoder hidden states of unconditioning"""
+        self.text_uncond = text_uncond
+        """ Encoder hidden states of text conditioning from negative prompt"""
 
 
 class CFGDenoisedParams:
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index ea974be04..93f0e55a0 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -106,8 +106,8 @@ class CFGDenoiser(torch.nn.Module):
         x_in = denoiser_params.x
         image_cond_in = denoiser_params.image_cond
         sigma_in = denoiser_params.sigma
-        tensor = denoiser_params.tensor
-        uncond = denoiser_params.uncond
+        tensor = denoiser_params.text_cond
+        uncond = denoiser_params.text_uncond
 
         if tensor.shape[1] == uncond.shape[1]:
             if not is_edit_model: