aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorlaksjdjf <aikasama@hotmail.co.jp>2023-02-24 14:03:46 +0900
committerGitHub <noreply@github.com>2023-02-24 14:03:46 +0900
commit327186b484c344598522a989a4b4859d6b90fb04 (patch)
tree83be7d6b168242796d2b304a2e9520409ac0bd8d
parent0cc0ee1bcb4c24a8c9715f66cede06601bfc00c8 (diff)
Update script_callbacks.py
-rw-r--r--modules/script_callbacks.py8
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: