Latent

Module: generative_models.models.diffusion.latent

Source: generative_models/models/diffusion/latent.py

Overview

Latent Diffusion Model implementation.

Classes

LDMModel

class LDMModel

SimpleDecoder

class SimpleDecoder

SimpleEncoder

class SimpleEncoder

Functions

call

def __call__()

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

init

def __init__()

decode

def decode()

encode

def encode()

loss

def loss()

output_dim

def output_dim()

reparameterize

def reparameterize()

sample

def sample()

Module Statistics

• Classes: 3
• Functions: 12
• Imports: 5