Commit 7f200512 authored by AUTOMATIC's avatar AUTOMATIC

rename CFGDenoiserParams fields for #8064

parent af416a2d
......@@ -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:
......
......@@ -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:
......
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