From 4c94f0f590b17830102d7eda6116a62f39497ac8 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 22 Oct 2023 08:08:36 -0700 Subject: [PATCH] reproduce previous results --- fooocus_version.py | 2 +- modules/default_pipeline.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fooocus_version.py b/fooocus_version.py index adf222ab..175d3fed 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.1.728' +version = '2.1.729' diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py index b30f32ad..fe2754b3 100644 --- a/modules/default_pipeline.py +++ b/modules/default_pipeline.py @@ -281,10 +281,10 @@ def vae_parse(latent): @torch.no_grad() @torch.inference_mode() -def noise_parse(latent: torch.Tensor, seed: int, noise_inds=None): +def noise_parse(latent: torch.Tensor, seed: int, noise_inds=None, k=0.9): noise = fcbh.sample.prepare_noise(latent, seed=seed, noise_inds=noise_inds) - s, m = torch.std_mean(latent, dim=1, keepdim=True) - return m + s * noise + offset = torch.mean(latent, dim=1, keepdim=True) + return offset * k + noise @torch.no_grad()