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

Update script_callbacks.py

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

8
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):
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"""
@ -45,6 +45,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:
def __init__(self, x, sampling_step, total_sampling_steps):