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 Original line Diff line number Diff line
@@ -29,7 +29,7 @@ class ImageSaveParams:




class CFGDenoiserParams:
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
        self.x = x
        """Latent image representation in the process of being denoised"""
        """Latent image representation in the process of being denoised"""
        
        
@@ -45,6 +45,12 @@ class CFGDenoiserParams:
        self.total_sampling_steps = total_sampling_steps
        self.total_sampling_steps = total_sampling_steps
        """Total number of sampling steps planned"""
        """Total number of sampling steps planned"""
        
        
        self.tensor = tensor
        """ Encoder hidden states of condtioning"""
        
        self.uncond = uncond
        """ Encoder hidden states of unconditioning"""



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