Commit cd88e21d authored by DepFA's avatar DepFA Committed by GitHub

Class Name typo and add descriptions to fields.

parent 8ae0ea9d
...@@ -24,13 +24,22 @@ class ImageSaveParams: ...@@ -24,13 +24,22 @@ class ImageSaveParams:
"""dictionary with parameters for image's PNG info data; infotext will have the key 'parameters'""" """dictionary with parameters for image's PNG info data; infotext will have the key 'parameters'"""
class CGFDenoiserParams: class CFGDenoiserParams:
def __init__(self, x_in, image_cond_in, sigma_in, sampling_step, total_sampling_steps): def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
self.x_in = x_in self.x = x
self.image_cond_in = image_cond_in """Latent image representation in the process of being denoised"""
self.sigma_in = sigma_in
self.image_cond = image_cond
"""Conditioning image"""
self.sigma = sigma
"""Current sigma noise step value"""
self.sampling_step = sampling_step self.sampling_step = sampling_step
"""Current Sampling step number"""
self.total_sampling_steps = total_sampling_steps self.total_sampling_steps = total_sampling_steps
"""Total number of sampling steps planned"""
ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"]) ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"])
...@@ -94,7 +103,7 @@ def image_saved_callback(params: ImageSaveParams): ...@@ -94,7 +103,7 @@ def image_saved_callback(params: ImageSaveParams):
report_exception(c, 'image_saved_callback') report_exception(c, 'image_saved_callback')
def cfg_denoiser_callback(params: CGFDenoiserParams): def cfg_denoiser_callback(params: CFGDenoiserParams):
for c in callbacks_cfg_denoiser: for c in callbacks_cfg_denoiser:
try: try:
c.callback(params) c.callback(params)
...@@ -153,7 +162,7 @@ def on_image_saved(callback): ...@@ -153,7 +162,7 @@ def on_image_saved(callback):
def on_cfg_denoiser(callback): def on_cfg_denoiser(callback):
"""register a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs. """register a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs.
The callback is called with one argument: The callback is called with one argument:
- params: CGFDenoiserParams - parameters to be passed to the inner model and sampling state details. - params: CFGDenoiserParams - parameters to be passed to the inner model and sampling state details.
""" """
add_callback(callbacks_cfg_denoiser, callback) add_callback(callbacks_cfg_denoiser, 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