MCPcopy Create free account

hub / github.com/NVlabs/Sana / functions

Functions2,606 in github.com/NVlabs/Sana

⨍ Functions 2,606◇ Types & classes 463↳ Endpoints 11
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads, num_buckets, shared_pos=True, dropout=0.1)
diffusion/model/wan/t5.py:137
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads, num_buckets, shared_pos=True, dropout=0.1)
diffusion/model/wan/t5.py:161
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
diffusion/model/wan/t5.py:188
Method__init__
(self, vocab, dim, dim_attn, dim_ffn, num_heads, num_layers, num_buckets, shared_pos=True, dropout=0.1)
diffusion/model/wan/t5.py:233
Method__init__
(self, vocab, dim, dim_attn, dim_ffn, num_heads, num_layers, num_buckets, shared_pos=True, dropout=0.1)
diffusion/model/wan/t5.py:270
Method__init__
( self, text_len, dtype=torch.bfloat16, device=None, checkpoint_path=N
diffusion/model/wan/t5.py:421
Method__init__
(self, *args, **kwargs)
diffusion/model/wan/vae.py:27
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
diffusion/model/wan/vae.py:44
Method__init__
(self, dim, mode)
diffusion/model/wan/vae.py:67
Method__init__
(self, in_dim, out_dim, dropout=0.0)
diffusion/model/wan/vae.py:166
Method__init__
(self, dim)
diffusion/model/wan/vae.py:207
Method__init__
( self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4], num_res_blocks=2,
diffusion/model/wan/vae.py:242
Method__init__
( self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4], num_res_blocks=2,
diffusion/model/wan/vae.py:340
Method__init__
( self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4], num_res_blocks=2,
diffusion/model/wan/vae.py:447
Method__init__
(self, *args, **kwargs)
diffusion/model/wan/model_wrapper.py:32
Method__init__
(self, dim, num_heads, dropout=0.1, eps=1e-5)
diffusion/model/wan/xlm_roberta.py:11
Method__init__
(self, dim, num_heads, post_norm, dropout=0.1, eps=1e-5)
diffusion/model/wan/xlm_roberta.py:49
Method__init__
(self, device)
diffusion/model/wan/model.py:23
Method__init__
(self, device, detach=True, score_only="cos")
diffusion/model/wan/model.py:46
Method__init__
(self, dim, eps=1e-5)
diffusion/model/wan/model.py:167
Method__init__
(self, dim, eps=1e-6, elementwise_affine=False)
diffusion/model/wan/model.py:185
Method__init__
(self, dim, num_heads, window_size=(-1, -1), qk_norm=True, eps=1e-6, **kwargs)
diffusion/model/wan/model.py:197
Method__init__
(self, dim, num_heads, window_size=(-1, -1), qk_norm=True, eps=1e-6, **kwargs)
diffusion/model/wan/model.py:260
Method__init__
(self, in_dim, out_dim, kernel_size=3, stride=1, padding=1)
diffusion/model/wan/model.py:365
Method__init__
(self, dim, num_heads, window_size=(-1, -1), qk_norm=True, eps=1e-6, **kwargs)
diffusion/model/wan/model.py:381
Method__init__
(self, dim, num_heads, window_size=(-1, -1), qk_norm=True, eps=1e-6, **kwargs)
diffusion/model/wan/model.py:433
Method__init__
(self, dim, num_heads, window_size=(-1, -1), qk_norm=True, eps=1e-6)
diffusion/model/wan/model.py:516
Method__init__
( self, cross_attn_type, dim, ffn_dim, num_heads, window_size=
diffusion/model/wan/model.py:568
Method__init__
(self, in_dim, out_dim)
diffusion/model/wan/model.py:730
Method__init__
r""" Initialize the diffusion model backbone. Args: model_type (`str`, *optional*, defaults to 't2v'): Mo
diffusion/model/wan/model.py:756
Method__init__
( self, model_type="t2v", patch_size=(1, 2, 2), text_len=512, in_dim=1
diffusion/model/wan/model.py:1136
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0.0,
diffusion/model/nets/sana_multi_scale_adaln.py:51
Method__init__
(self, d_model, num_heads, attn_drop=0.0, proj_drop=0.0, qk_norm=False, **block_kwargs)
diffusion/model/nets/sana_blocks.py:49
Method__init__
(self, d_model, num_heads, attn_drop=0.0, proj_drop=0.0, qk_norm=False, **block_kwargs)
diffusion/model/nets/sana_blocks.py:102
Method__init__
( self, in_dim: int, out_dim: int, heads: Optional[int] = None, heads_
diffusion/model/nets/sana_blocks.py:309
Method__init__
( self, in_dim: int, out_dim: int, heads: Optional[int] = None, heads_
diffusion/model/nets/sana_blocks.py:394
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/sana_blocks.py:499
Method__init__
(self, attn)
diffusion/model/nets/sana_blocks.py:578
Method__init__
(self, attn)
diffusion/model/nets/sana_blocks.py:630
Method__init__
(self, attn)
diffusion/model/nets/sana_blocks.py:681
Method__init__
(self, attn)
diffusion/model/nets/sana_blocks.py:723
Method__init__
Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. qkv_bias (bool: I
diffusion/model/nets/sana_blocks.py:771
Method__init__
(self, hidden_size, patch_size, out_channels)
diffusion/model/nets/sana_blocks.py:880
Method__init__
(self, hidden_size, patch_size, out_channels)
diffusion/model/nets/sana_blocks.py:898
Method__init__
(self, final_hidden_size, c_emb_size, patch_size, out_channels)
diffusion/model/nets/sana_blocks.py:933
Method__init__
(self, hidden_size, decoder_hidden_size)
diffusion/model/nets/sana_blocks.py:951
Method__init__
(self, hidden_size, frequency_embedding_size=256)
diffusion/model/nets/sana_blocks.py:972
Method__init__
(self, hidden_size, frequency_embedding_size=256)
diffusion/model/nets/sana_blocks.py:1020
Method__init__
(self, num_classes, hidden_size, dropout_prob)
diffusion/model/nets/sana_blocks.py:1057
Method__init__
( self, in_channels, hidden_size, uncond_prob, act_layer=nn.GELU(appro
diffusion/model/nets/sana_blocks.py:1088
Method__init__
(self, in_channels, hidden_size, uncond_prob, act_layer=nn.GELU(approximate="tanh"), token_num=120)
diffusion/model/nets/sana_blocks.py:1146
Method__init__
(self, in_channels: int, out_channels: int)
diffusion/model/nets/sana_blocks.py:1179
Method__init__
( self, img_size=224, patch_size=16, in_chans=3, embed_dim=768,
diffusion/model/nets/sana_blocks.py:1194
Method__init__
( self, patch_size=16, in_chans=3, embed_dim=768, kernel_size=None,
diffusion/model/nets/sana_blocks.py:1238
Method__init__
( self, patch_size=(1, 2, 2), in_chans=3, embed_dim=768, kernel_size=N
diffusion/model/nets/sana_blocks.py:1274
Method__init__
(self, theta: int, axes_dim: List[int])
diffusion/model/nets/sana_blocks.py:1309
Method__init__
( self, attention_head_dim: int, patch_size: Tuple[int, int, int], max_seq_len
diffusion/model/nets/sana_blocks.py:1362
Method__init__
( self, attention_head_dim: int, patch_size: Tuple[int, int, int], max_seq_len: int, theta: float = 10
diffusion/model/nets/sana_blocks.py:1448
Method__init__
Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. qkv_bias (bool): I
diffusion/model/nets/sana_blocks.py:1608
Method__init__
(self, *args, chunk_size=200_000, **kwargs)
diffusion/model/nets/sana_blocks.py:1793
Method__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, di
diffusion/model/nets/basic_modules.py:31
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
diffusion/model/nets/basic_modules.py:110
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
diffusion/model/nets/basic_modules.py:201
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/basic_modules.py:344
Method__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, mi
diffusion/model/nets/basic_modules.py:387
Method__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
diffusion/model/nets/basic_modules.py:477
Method__init__
( self, in_dim: int, out_dim: int, use_bias=True, norm=None, a
diffusion/model/nets/basic_modules_linear.py:35
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
diffusion/model/nets/basic_modules_linear.py:148
Method__init__
( self, in_features: int, hidden_features: int, out_feature=None, kern
diffusion/model/nets/basic_modules_linear.py:246
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0, qk
diffusion/model/nets/sana.py:59
Method__init__
(self, input_dim, hidden_size, act_layer=nn.GELU)
diffusion/model/nets/sana_multi_scale_video_camctrl.py:77
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0.0,
diffusion/model/nets/sana_multi_scale_video_camctrl.py:108
Method__init__
( self, input_size=32, patch_size=(1, 2, 2), in_channels=4, hidden_siz
diffusion/model/nets/sana_multi_scale_video_camctrl.py:625
Method__init__
(self, base_block: SanaMSBlock, block_index: int)
diffusion/model/nets/sana_multi_scale_controlnet.py:43
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0.0,
diffusion/model/nets/sana_multi_scale_video.py:71
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0.0,
diffusion/model/nets/sana_U_shape_multi_scale.py:50
Method__init__
(self, pretrained_model: SanaMSCM, is_multiscale=False, head_block_ids=None)
diffusion/model/nets/sana_ladd.py:28
Method__init__
(self, *args, use_autograd_kernel: bool = False, **kwargs)
diffusion/model/nets/sana_gdn_blocks_triton.py:227
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0, in
diffusion/model/nets/sana_U_shape.py:55
Method__init__
(self, *args: object, conv_kernel_size: int = 0, **kwargs: object)
diffusion/model/nets/sana_gdn_blocks.py:1087
Method__init__
( self, hidden_size, num_heads, mlp_ratio=4.0, drop_path=0.0,
diffusion/model/nets/sana_multi_scale.py:59
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/sana_multi_scale.py:489
Method__init__
(self, fn: Callable)
diffusion/model/nets/ladd_blocks.py:26
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/ladd_blocks.py:35
Method__init__
(self, num_features: int, affine: bool = True, virtual_bs: int = 8, eps: float = 1e-5)
diffusion/model/nets/ladd_blocks.py:41
Method__init__
(self, *args, conv_kernel_size: int = 0, **kwargs)
diffusion/model/nets/sana_gdn_camctrl_blocks.py:543
Method__init__
(self, *args, conv_kernel_size: int = 0, **kwargs)
diffusion/model/nets/sana_gdn_camctrl_blocks.py:755
Method__init__
( self, dim: int, num_heads: int, eps=1e-15, use_bias=False, )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_fwd.py:86
Method__init__
( self, in_dim: int, out_dim: int, heads: Optional[int] = None, heads_
diffusion/model/nets/fastlinear/modules/lite_mla.py:30
Method__init__
(self, dim: int, num_heads: int)
diffusion/model/nets/fastlinear/modules/flash_attn.py:24
Method__init__
( self, dim: int, num_heads: int, eps=1e-15, use_bias=False, )
diffusion/model/nets/fastlinear/modules/triton_lite_mla.py:105
Method__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, mi
diffusion/model/nets/fastlinear/modules/triton_mb_conv_pre_glu.py:29
Method__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, mi
diffusion/model/nets/fastlinear/modules/mb_conv_pre_glu.py:27
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/fastlinear/modules/utils/custom_autotune.py:28
Method__init__
(self, *args, **kwargs)
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/custom_autotune.py:28
Method__init__
Initialize the RMSNorm normalization layer. Args: dim (int): The dimension of the input tensor.
diffusion/model/nets/fastlinear/modules/nn/norm.py:189
Method__init__
( self, in_dim: int, out_dim: int, kernel_size=3, stride=1, di
diffusion/model/nets/fastlinear/modules/nn/conv.py:26
Method__init__
(self, model_name: str)
diffusion/model/dc_ae/efficientvit/ae_model_zoo.py:78
Method__init__
(self, model_name: str)
diffusion/model/dc_ae/efficientvit/ae_model_zoo.py:95
Method__init__
(self, is_distributed=True)
diffusion/model/dc_ae/efficientvit/apps/utils/metric.py:29
← previousnext →1,601–1,700 of 2,606, ranked by callers