Dit

Module: generative_models.models.diffusion.dit

Source: generative_models/models/diffusion/dit.py

Overview

DiT (Diffusion Transformer) Model implementation.

Integrates the Diffusion Transformer backbone with the diffusion framework.

Classes

DiTModel

class DiTModel

Functions

call

def __call__()

init

def __init__()

create_dit_backbone

def create_dit_backbone()

generate

def generate()

sample_step

def sample_step()

Module Statistics

• Classes: 1
• Functions: 5
• Imports: 7