aboutsummaryrefslogtreecommitdiff
path: root/modules/script_callbacks.py
diff options
context:
space:
mode:
authorInvincibleDude <81354513+InvincibleDude@users.noreply.github.com>2023-03-14 16:55:59 +0300
committerGitHub <noreply@github.com>2023-03-14 16:55:59 +0300
commitf5e44364535ccc7efef445dacb6395c3942f2f17 (patch)
treeca7384e6225fccdae1e89db1e0ccc44dd60c7311 /modules/script_callbacks.py
parentf6e27378404631d951656388fc178b784fe1495b (diff)
parenta9fed7c364061ae6efb37f797b6b522cb3cf7aa2 (diff)
Merge branch 'master' into improved-hr-conflict-test
Diffstat (limited to 'modules/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..07911876 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, text_cond, text_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.text_cond = text_cond
+ """ Encoder hidden states of text conditioning from prompt"""
+
+ self.text_uncond = text_uncond
+ """ Encoder hidden states of text conditioning from negative prompt"""
class CFGDenoisedParams: