Function dist_sample_deterministic

lib_layerdiffuse/vae.py:229–232

(dist: DiagonalGaussianDistribution, perturbation: torch.Tensor)

def dist_sample_deterministic(dist: DiagonalGaussianDistribution, perturbation: torch.Tensor):
    # Modified from diffusers.models.autoencoders.vae.DiagonalGaussianDistribution.sample()
    x = dist.mean + dist.std * perturbation.to(dist.std)
    return x

class TransparentVAE(torch.nn.Module):
    def __init__(self, sd_vae, dtype=torch.float16, encoder_file=None, decoder_file=None, alpha=300.0, latent_c=16, *args, **kwargs):