Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Vchitect/SEINE
/ types & classes
Types & classes
42 in github.com/Vchitect/SEINE
⨍
Functions
203
◇
Types & classes
42
↓ 7 callers
Class
InflatedConv3d
models/resnet.py:13
↓ 6 callers
Class
ResnetBlock3D
models/resnet.py:113
↓ 3 callers
Class
Transformer3DModel
models/attention.py:314
↓ 2 callers
Class
CrossAttention
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 callers
Class
Downsample3D
models/resnet.py:79
↓ 2 callers
Class
TextEmbedder
Embeds text prompt into vector representations. Also handles text dropout for classifier-free guidance.
models/clip.py:61
↓ 2 callers
Class
Upsample3D
models/resnet.py:24
↓ 1 callers
Class
BasicTransformerBlock
models/attention.py:439
↓ 1 callers
Class
CrossAttnDownBlock3D
models/unet_blocks.py:235
↓ 1 callers
Class
CrossAttnUpBlock3D
models/unet_blocks.py:444
↓ 1 callers
Class
DownBlock3D
models/unet_blocks.py:365
↓ 1 callers
Class
FrozenCLIPEmbedder
Uses the CLIP transformer encoder for text (from Hugging Face)
models/clip.py:32
↓ 1 callers
Class
GaussianDiffusion
Utilities for training and sampling diffusion models. Original ported from this codebase: https://github.com/hojonathanho/diffusion/bl
diffusion/gaussian_diffusion.py:147
↓ 1 callers
Class
GroupNorm32
models/utils.py:137
↓ 1 callers
Class
LossSecondMomentResampler
diffusion/timestep_sampler.py:120
↓ 1 callers
Class
Mish
models/resnet.py:210
↓ 1 callers
Class
RelativePositionBias
models/unet.py:53
↓ 1 callers
Class
RelativePositionBias
models/attention.py:930
↓ 1 callers
Class
SpacedDiffusion
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 callers
Class
TemporalAttention
models/attention.py:797
↓ 1 callers
Class
Transformer3DModelOutput
models/attention.py:29
↓ 1 callers
Class
UNet3DConditionOutput
models/unet.py:94
↓ 1 callers
Class
UNetMidBlock3DCrossAttn
models/unet_blocks.py:145
↓ 1 callers
Class
UniformSampler
diffusion/timestep_sampler.py:62
↓ 1 callers
Class
UpBlock3D
models/unet_blocks.py:577
↓ 1 callers
Class
_WrappedModel
diffusion/respace.py:118
Class
AbstractEncoder
models/clip.py:24
Class
CenterCropResizeVideo
First use the short side for cropping length, center crop video, then resize to the specified size
datasets/video_transforms.py:230
Class
CenterCropVideo
datasets/video_transforms.py:268
Class
CheckpointFunction
models/utils.py:42
Class
LossAwareSampler
diffusion/timestep_sampler.py:71
Class
LossType
diffusion/gaussian_diffusion.py:46
Class
ModelMeanType
Which type of output the model predicts.
diffusion/gaussian_diffusion.py:23
Class
ModelVarType
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
Class
NormalizeVideo
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
Class
RandomCropVideo
datasets/video_transforms.py:194
Class
ResizeVideo
First use the short side for cropping length, center crop video, then resize to the specified size
datasets/video_transforms.py:346
Class
ScheduleSampler
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
Class
SiLU
models/utils.py:132
Class
SparseCausalAttention
models/attention.py:650
Class
ToTensorVideo
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
Class
UNet3DConditionModel
models/unet.py:98