MCPcopy Create free account

hub / github.com/Vchitect/SEINE / types & classes

Types & classes42 in github.com/Vchitect/SEINE

↓ 7 callersClassInflatedConv3d
models/resnet.py:13
↓ 6 callersClassResnetBlock3D
models/resnet.py:113
↓ 3 callersClassTransformer3DModel
models/attention.py:314
↓ 2 callersClassCrossAttention
r""" copy from diffuser 0.11.1 A cross attention layer. Parameters: query_dim (`int`): The number of channels in the query.
models/attention.py:43
↓ 2 callersClassDownsample3D
models/resnet.py:79
↓ 2 callersClassTextEmbedder
Embeds text prompt into vector representations. Also handles text dropout for classifier-free guidance.
models/clip.py:61
↓ 2 callersClassUpsample3D
models/resnet.py:24
↓ 1 callersClassBasicTransformerBlock
models/attention.py:439
↓ 1 callersClassCrossAttnDownBlock3D
models/unet_blocks.py:235
↓ 1 callersClassCrossAttnUpBlock3D
models/unet_blocks.py:444
↓ 1 callersClassDownBlock3D
models/unet_blocks.py:365
↓ 1 callersClassFrozenCLIPEmbedder
Uses the CLIP transformer encoder for text (from Hugging Face)
models/clip.py:32
↓ 1 callersClassGaussianDiffusion
Utilities for training and sampling diffusion models. Original ported from this codebase: https://github.com/hojonathanho/diffusion/bl
diffusion/gaussian_diffusion.py:147
↓ 1 callersClassGroupNorm32
models/utils.py:137
↓ 1 callersClassLossSecondMomentResampler
diffusion/timestep_sampler.py:120
↓ 1 callersClassMish
models/resnet.py:210
↓ 1 callersClassRelativePositionBias
models/unet.py:53
↓ 1 callersClassRelativePositionBias
models/attention.py:930
↓ 1 callersClassSpacedDiffusion
A diffusion process which can skip steps in a base diffusion process. :param use_timesteps: a collection (sequence or set) of timesteps fro
diffusion/respace.py:65
↓ 1 callersClassTemporalAttention
models/attention.py:797
↓ 1 callersClassTransformer3DModelOutput
models/attention.py:29
↓ 1 callersClassUNet3DConditionOutput
models/unet.py:94
↓ 1 callersClassUNetMidBlock3DCrossAttn
models/unet_blocks.py:145
↓ 1 callersClassUniformSampler
diffusion/timestep_sampler.py:62
↓ 1 callersClassUpBlock3D
models/unet_blocks.py:577
↓ 1 callersClass_WrappedModel
diffusion/respace.py:118
ClassAbstractEncoder
models/clip.py:24
ClassCenterCropResizeVideo
First use the short side for cropping length, center crop video, then resize to the specified size
datasets/video_transforms.py:230
ClassCenterCropVideo
datasets/video_transforms.py:268
ClassCheckpointFunction
models/utils.py:42
ClassLossAwareSampler
diffusion/timestep_sampler.py:71
ClassLossType
diffusion/gaussian_diffusion.py:46
ClassModelMeanType
Which type of output the model predicts.
diffusion/gaussian_diffusion.py:23
ClassModelVarType
What is used as the model's output variance. The LEARNED_RANGE option has been added to allow the model to predict values between FIXE
diffusion/gaussian_diffusion.py:33
ClassNormalizeVideo
Normalize the video clip by mean subtraction and division by standard deviation Args: mean (3-tuple): pixel RGB mean std
datasets/video_transforms.py:299
ClassRandomCropVideo
datasets/video_transforms.py:194
ClassResizeVideo
First use the short side for cropping length, center crop video, then resize to the specified size
datasets/video_transforms.py:346
ClassScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unb
diffusion/timestep_sampler.py:27
ClassSiLU
models/utils.py:132
ClassSparseCausalAttention
models/attention.py:650
ClassToTensorVideo
Convert tensor data type from uint8 to float, divide value by 255.0 and permute the dimensions of clip tensor
datasets/video_transforms.py:324
ClassUNet3DConditionModel
models/unet.py:98