michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

rename CFGDenoiserParams fields for #8064

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/script_callbacks.py
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:

4
modules/sd_samplers_kdiffusion.py
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: