diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-08 07:23:08 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-08 07:23:08 +0000 |
| commit | 6869d95890849c9b209bb66774539bfdf870df2c (patch) | |
| tree | 015d7e7fb4813dd9276bec2d71f9318e508a1669 /modules/script_callbacks.py | |
| parent | 2f98a35fc4508494355c01ec45f5bec725f570a6 (diff) | |
| parent | 37906e429ae5a92f9ea96ab6dc2157b1d7c4d8b6 (diff) | |
| download | stable-diffusion-webui-gfx803-6869d95890849c9b209bb66774539bfdf870df2c.tar.gz stable-diffusion-webui-gfx803-6869d95890849c9b209bb66774539bfdf870df2c.tar.bz2 stable-diffusion-webui-gfx803-6869d95890849c9b209bb66774539bfdf870df2c.zip | |
Merge pull request #14573 from continue-revolution/conrevo/add-p-to-cfg
Add self to CFGDenoiserParams
Diffstat (limited to 'modules/script_callbacks.py')
| -rw-r--r-- | modules/script_callbacks.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py index 9ed7ad21..a54cb3eb 100644 --- a/modules/script_callbacks.py +++ b/modules/script_callbacks.py @@ -41,7 +41,7 @@ class ExtraNoiseParams: class CFGDenoiserParams:
- def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, text_cond, text_uncond):
+ def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, text_cond, text_uncond, denoiser=None):
self.x = x
"""Latent image representation in the process of being denoised"""
@@ -63,6 +63,9 @@ class CFGDenoiserParams: self.text_uncond = text_uncond
""" Encoder hidden states of text conditioning from negative prompt"""
+ self.denoiser = denoiser
+ """Current CFGDenoiser object with processing parameters"""
+
class CFGDenoisedParams:
def __init__(self, x, sampling_step, total_sampling_steps, inner_model):
|