diff options
| author | wangshuai09 <391746016@qq.com> | 2024-01-30 11:15:41 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-30 11:15:41 +0000 |
| commit | 74ff85a1a1ee4cce432b1c7d33c1eda831f68d48 (patch) | |
| tree | 99b70e0fef8422c8f603bf7faa1a393091cb2a8b /modules/script_callbacks.py | |
| parent | ec124607f47371a6cfd61a795f86a7f1cbd44651 (diff) | |
| parent | ce168ab5dbc8b54b7245f352a2eaa55a37019b91 (diff) | |
| download | stable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.tar.gz stable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.tar.bz2 stable-diffusion-webui-gfx803-74ff85a1a1ee4cce432b1c7d33c1eda831f68d48.zip | |
Merge branch 'dev' into npu_support
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):
|