Update script_callbacks.py

This commit is contained in:
laksjdjf 2023-02-24 14:03:46 +09:00 committed by GitHub
parent 0cc0ee1bcb
commit 327186b484
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions

View File

@ -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"""
@ -44,6 +44,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: