diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index edd0e2a72..c5bb3e71e 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):
+    def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, tensor, uncond):
         self.x = x
         """Latent image representation in the process of being denoised"""
         
@@ -44,6 +44,12 @@ class CFGDenoiserParams:
         
         self.total_sampling_steps = total_sampling_steps
         """Total number of sampling steps planned"""
+        
+        self.tensor = tensor
+        """ Encoder hidden states of condtioning"""
+        
+        self.uncond = uncond
+        """ Encoder hidden states of unconditioning"""
 
 
 class CFGDenoisedParams: