rename CFGDenoiserParams fields for #8064

This commit is contained in:
AUTOMATIC 2023-03-11 14:52:29 +03:00
parent af416a2dbd
commit 7f2005127f
2 changed files with 7 additions and 7 deletions

View File

@ -29,7 +29,7 @@ class ImageSaveParams:
class CFGDenoiserParams:
def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, tensor, uncond):
def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps, text_cond, text_uncond):
self.x = x
"""Latent image representation in the process of being denoised"""
@ -45,11 +45,11 @@ class CFGDenoiserParams:
self.total_sampling_steps = total_sampling_steps
"""Total number of sampling steps planned"""
self.tensor = tensor
""" Encoder hidden states of conditioning"""
self.text_cond = text_cond
""" Encoder hidden states of text conditioning from prompt"""
self.uncond = uncond
""" Encoder hidden states of unconditioning"""
self.text_uncond = text_uncond
""" Encoder hidden states of text conditioning from negative prompt"""
class CFGDenoisedParams:

View File

@ -106,8 +106,8 @@ class CFGDenoiser(torch.nn.Module):
x_in = denoiser_params.x
image_cond_in = denoiser_params.image_cond
sigma_in = denoiser_params.sigma
tensor = denoiser_params.tensor
uncond = denoiser_params.uncond
tensor = denoiser_params.text_cond
uncond = denoiser_params.text_uncond
if tensor.shape[1] == uncond.shape[1]:
if not is_edit_model: