From 13ec2053764dd923422c01ba7494ebd324601e92 Mon Sep 17 00:00:00 2001 From: virginiafdez Date: Wed, 1 Feb 2023 11:45:46 +0000 Subject: [PATCH] Modify sample function to divide by scale factor before passing to the VAE for decoding. --- generative/inferers/inferer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generative/inferers/inferer.py b/generative/inferers/inferer.py index 6927f5ca..f65bdb20 100644 --- a/generative/inferers/inferer.py +++ b/generative/inferers/inferer.py @@ -347,14 +347,14 @@ def sample( latent = outputs with torch.no_grad(): - image = autoencoder_model.decode_stage_2_outputs(latent) * self.scale_factor + image = autoencoder_model.decode_stage_2_outputs(latent / self.scale_factor) if save_intermediates: intermediates = [] for latent_intermediate in latent_intermediates: with torch.no_grad(): intermediates.append( - autoencoder_model.decode_stage_2_outputs(latent_intermediate) * self.scale_factor + autoencoder_model.decode_stage_2_outputs(latent_intermediate / self.scale_factor) ) return image, intermediates