diff options
| -rw-r--r-- | modules/script_callbacks.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py index edd0e2a7..c5bb3e71 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:
|