StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update script_callbacks.py

Browse Source
This commit is contained in:
Sakura-Luna 2023-05-14 12:56:34 +08:00 committed by GitHub
parent 3078001439
commit 8abfc95013
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/script_callbacks.py
View File

@ -64,7 +64,7 @@ class CFGDenoisedParams:
"""Total number of sampling steps planned""" """Total number of sampling steps planned"""
self.inner_model = inner_model self.inner_model = inner_model
"""Inner model reference that is being used for denoising""" """Inner model reference used for denoising"""
class AfterCFGCallbackParams: class AfterCFGCallbackParams:
@ -79,7 +79,7 @@ class AfterCFGCallbackParams:
"""Total number of sampling steps planned""" """Total number of sampling steps planned"""
self.output_altered = False self.output_altered = False
"""A flag for CFGDenoiser that indicates whether the output has been altered by the callback""" """A flag for CFGDenoiser indicating whether the output has been altered by the callback"""
class UiTrainTabParams: class UiTrainTabParams:
@ -360,9 +360,9 @@ def on_cfg_denoised(callback):
def on_cfg_after_cfg(callback): def on_cfg_after_cfg(callback):
"""register a function to be called in the kdiffussion cfg_denoiser method after cfg calculations has completed. """register a function to be called in the kdiffussion cfg_denoiser method after cfg calculations are completed.
The callback is called with one argument: The callback is called with one argument:
- params: CFGDenoisedParams - parameters to be passed to the inner model and sampling state details. - params: AfterCFGCallbackParams - parameters to be passed to the script for post-processing after cfg calculation.
""" """
add_callback(callback_map['callbacks_cfg_after_cfg'], callback) add_callback(callback_map['callbacks_cfg_after_cfg'], callback)