modelzoo.vision.pytorch.dit.model.DiTModel#

class modelzoo.vision.pytorch.dit.model.DiTModel[source]#

Bases: torch.nn.Module

Methods

build_model

forward

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

Call self as a function.

__init__(params)[source]#
static __new__(cls, *args: Any, **kwargs: Any) Any#