lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Gradio 4 + WebUI 1.10

Browse Source
This commit is contained in:
layerdiffusion
2024-07-26 08:51:34 -07:00
parent e95333c556
commit e26abf87ec
201 changed files with 7562 additions and 4834 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm_patched/ldm/models/autoencoder.py
View File

@@ -182,7 +182,7 @@ class AutoencodingEngineLegacy(AutoencodingEngine):
return params
def encode(
self, x: torch.Tensor, return_reg_log: bool = False
self, x: torch.Tensor, regulation=None, return_reg_log: bool = False
) -> Union[torch.Tensor, Tuple[torch.Tensor, dict]]:
if self.max_batch_size is None:
z = self.encoder(x)
@@ -198,7 +198,7 @@ class AutoencodingEngineLegacy(AutoencodingEngine):
z.append(z_batch)
z = torch.cat(z, 0)
z, reg_log = self.regularization(z)
z, reg_log = self.regularization(z) if regulation is None else regulation(z)
if return_reg_log:
return z, reg_log
return z