openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-15 15:13:45 +03:00
Code Issues Packages Projects Releases 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):