Commit 8abfc950 authored by Sakura-Luna's avatar Sakura-Luna Committed by GitHub

Update script_callbacks.py

parent 30780014
......@@ -64,7 +64,7 @@ class CFGDenoisedParams:
"""Total number of sampling steps planned"""
self.inner_model = inner_model
"""Inner model reference that is being used for denoising"""
"""Inner model reference used for denoising"""
class AfterCFGCallbackParams:
......@@ -79,7 +79,7 @@ class AfterCFGCallbackParams:
"""Total number of sampling steps planned"""
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:
......@@ -360,9 +360,9 @@ def on_cfg_denoised(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:
- 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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment