Unverified Commit 327186b4 authored by laksjdjf's avatar laksjdjf Committed by GitHub
Browse files

Update script_callbacks.py

parent 0cc0ee1b
Loading
Loading
Loading
Loading
+7 −1
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ class ImageSaveParams:


class CFGDenoiserParams:
    def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
    def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, tensor, uncond):
        self.x = x
        """Latent image representation in the process of being denoised"""
        
@@ -45,6 +45,12 @@ class CFGDenoiserParams:
        self.total_sampling_steps = total_sampling_steps
        """Total number of sampling steps planned"""
        
        self.tensor = tensor
        """ Encoder hidden states of condtioning"""
        
        self.uncond = uncond
        """ Encoder hidden states of unconditioning"""


class CFGDenoisedParams:
    def __init__(self, x, sampling_step, total_sampling_steps):