modelzoo.vision.pytorch.dit.model.DiTModel

class modelzoo.vision.pytorch.dit.model.DiTModel

Bases: torch.nn.Module

Methods

build_model
forward

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

__init__(params)

static __new__(cls, *args: Any, **kwargs: Any) → Any

previous

modelzoo.vision.pytorch.dit.model

next

modelzoo.vision.pytorch.dit.modeling_dit