MCPcopy Create free account

hub / github.com/Tencent-Hunyuan/HunyuanVideo / functions

Functions197 in github.com/Tencent-Hunyuan/HunyuanVideo

⨍ Functions 197◇ Types & classes 35
↓ 1 callersMethodtemporal_tiled_decode
(self, z: torch.FloatTensor, return_dict: bool = True)
hyvideo/vae/autoencoder_kl_causal_3d.py:500
↓ 1 callersMethodtemporal_tiled_encode
(self, x: torch.FloatTensor, return_dict: bool = True)
hyvideo/vae/autoencoder_kl_causal_3d.py:465
↓ 1 callersFunctiontest_mm_double_stream_block_attention
(rank, world_size)
tests/test_attention.py:44
↓ 1 callersFunctiontest_mm_single_stream_block_attention
(rank, world_size)
tests/test_attention.py:112
↓ 1 callersFunctiontimestep_embedding
Create sinusoidal timestep embeddings. Args: t (torch.Tensor): a 1-D Tensor of N indices, one per batch element. These may be fracti
hyvideo/modules/embed_layers.py:93
↓ 1 callersMethodunpatchify
x: (N, T, patch_size**2 * C) imgs: (N, H, W, C)
hyvideo/modules/models.py:697
Method__call__
r""" The call function to the pipeline for generation. Args: prompt (`str` or `List[str]`): The prompt or
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:664
Method__init__
( self, args, vae, vae_kwargs, text_encoder, model, te
hyvideo/inference.py:370
Method__init__
( self, in_channels: int = 3, out_channels: int = 3, up_block_types: Tuple
hyvideo/vae/vae.py:141
Method__init__
(self, parameters: torch.Tensor, deterministic: bool = False)
hyvideo/vae/vae.py:295
Method__init__
( self, channels: int, use_conv: bool = False, use_conv_transpose: bool =
hyvideo/vae/unet_causal_3d_blocks.py:82
Method__init__
( self, channels: int, use_conv: bool = False, out_channels: Optional[int]
hyvideo/vae/unet_causal_3d_blocks.py:190
Method__init__
( self, *, in_channels: int, out_channels: Optional[int] = None,
hyvideo/vae/unet_causal_3d_blocks.py:254
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0,
hyvideo/vae/unet_causal_3d_blocks.py:531
Method__init__
( self, in_channels: int, out_channels: int, dropout: float = 0.0,
hyvideo/vae/unet_causal_3d_blocks.py:632
Method__init__
( self, in_channels: int, out_channels: int, resolution_idx: Optional[int]
hyvideo/vae/unet_causal_3d_blocks.py:698
Method__init__
( self, in_channels: int = 3, out_channels: int = 3, down_block_types: Tup
hyvideo/vae/autoencoder_kl_causal_3d.py:64
Method__init__
( self, text_encoder_type: str, max_length: int, text_encoder_precision: Optio
hyvideo/text_encoder/__init__.py:103
Method__init__
( self, vae: AutoencoderKL, text_encoder: TextEncoder, transformer: HYVideoDif
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:169
Method__init__
( self, num_train_timesteps: int = 1000, shift: float = 1.0, reverse: bool = T
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:71
Method__init__
Initialize the RMSNorm normalization layer. Args: dim (int): The dimension of the input tensor. eps (float,
hyvideo/modules/norm_layers.py:6
Method__init__
(self, in_channels, hidden_size, act_layer, dtype=None, device=None)
hyvideo/modules/embed_layers.py:69
Method__init__
( self, hidden_size, act_layer, frequency_embedding_size=256, max_peri
hyvideo/modules/embed_layers.py:125
Method__init__
( self, hidden_size: int, heads_num: int, mlp_width_ratio: float = 4.0,
hyvideo/modules/models.py:263
Method__init__
( self, args: Any, patch_size: list = [1, 2, 2], in_channels: int = 4, # Shou
hyvideo/modules/models.py:449
Method__init__
( self, hidden_size, heads_num, mlp_width_ratio: str = 4.0, mlp_drop_r
hyvideo/modules/token_refiner.py:17
Method__init__
( self, hidden_size, heads_num, depth, mlp_width_ratio: float = 4.0,
hyvideo/modules/token_refiner.py:104
Method__init__
(self, in_dim: int, hidden_dim: int, device=None, dtype=None)
hyvideo/modules/mlp_layers.py:65
Method__init__
( self, hidden_size, patch_size, out_channels, act_layer, device=None, dtype=None )
hyvideo/modules/mlp_layers.py:79
Method__init__
( self, hidden_size: int, factor: int, act_layer: Callable, dtype=None
hyvideo/modules/modulate_layers.py:9
Method__len__
(self)
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:256
Method__repr__
(self)
hyvideo/text_encoder/__init__.py:197
Method_set_gradient_checkpointing
(self, module, value=False)
hyvideo/vae/autoencoder_kl_causal_3d.py:134
Method_sigma_to_t
(self, sigma)
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:122
Functionas_list_of_2tuple
(x)
hyvideo/utils/helpers.py:32
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
hyvideo/vae/autoencoder_kl_causal_3d.py:183
Methodbegin_index
The index for the first timestep. It should be set from pipeline with `set_begin_index` method.
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:105
Functionckpt_forward
(*inputs)
hyvideo/modules/modulate_layers.py:72
Functionckpt_wrapper
(module)
hyvideo/modules/modulate_layers.py:71
Methodclip_skip
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:639
Methodcreate_custom_forward
(module)
hyvideo/vae/vae.py:240
Methodcross_attention_kwargs
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:651
Methodcustom_forward
(*inputs)
hyvideo/vae/vae.py:241
Methoddecode_latents
(self, latents, enable_tiling=True)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:451
Methoddisable_deterministic
(self)
hyvideo/modules/models.py:323
Methoddisable_deterministic
(self)
hyvideo/modules/models.py:589
Methoddisable_slicing
r""" Disable sliced VAE decoding. If `enable_slicing` was previously enabled, this method will go back to computing decoding in one
hyvideo/vae/autoencoder_kl_causal_3d.py:174
Methoddisable_tiling
r""" Disable tiled VAE decoding. If `enable_tiling` was previously enabled, this method will go back to computing decoding in one st
hyvideo/vae/autoencoder_kl_causal_3d.py:159
Methoddo_classifier_free_guidance
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:646
Methodenable_deterministic
(self)
hyvideo/modules/models.py:320
Methodenable_deterministic
(self)
hyvideo/modules/models.py:583
Methodenable_slicing
r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in s
hyvideo/vae/autoencoder_kl_causal_3d.py:167
Methodfn_recursive_add_processors
(name: str, module: torch.nn.Module, processors: Dict[str, AttentionProcessor])
hyvideo/vae/autoencoder_kl_causal_3d.py:192
Methodfn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
hyvideo/vae/autoencoder_kl_causal_3d.py:230
Methodforward
r"""The forward method of the `EncoderCausal3D` class.
hyvideo/vae/vae.py:115
Methodforward
r"""The forward method of the `DecoderCausal3D` class.
hyvideo/vae/vae.py:227
Methodforward
(self, x)
hyvideo/vae/unet_causal_3d_blocks.py:72
Methodforward
( self, hidden_states: torch.FloatTensor, output_size: Optional[int] = None,
hyvideo/vae/unet_causal_3d_blocks.py:129
Methodforward
(self, hidden_states: torch.FloatTensor, scale: float = 1.0)
hyvideo/vae/unet_causal_3d_blocks.py:236
Methodforward
( self, input_tensor: torch.FloatTensor, temb: torch.FloatTensor, scale: f
hyvideo/vae/unet_causal_3d_blocks.py:348
Methodforward
(self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None)
hyvideo/vae/unet_causal_3d_blocks.py:615
Methodforward
(self, hidden_states: torch.FloatTensor, scale: float = 1.0)
hyvideo/vae/unet_causal_3d_blocks.py:686
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None, scale: float = 1.
hyvideo/vae/unet_causal_3d_blocks.py:754
Methodforward
r""" Args: sample (`torch.FloatTensor`): Input sample. sample_posterior (`bool`, *optional*, defaults to `False`):
hyvideo/vae/autoencoder_kl_causal_3d.py:533
Methodforward
( self, text, use_attention_mask=None, output_hidden_states=False, do_
hyvideo/text_encoder/__init__.py:340
Methodforward
Forward pass through the RMSNorm layer. Args: x (torch.Tensor): The input tensor. Returns: torch.Te
hyvideo/modules/norm_layers.py:45
Methodforward
(self, x)
hyvideo/modules/embed_layers.py:54
Methodforward
(self, caption)
hyvideo/modules/embed_layers.py:86
Methodforward
(self, t)
hyvideo/modules/embed_layers.py:152
Methodforward
( self, img: torch.Tensor, txt: torch.Tensor, vec: torch.Tensor, cu_se
hyvideo/modules/models.py:132
Methodforward
( self, x: torch.Tensor, vec: torch.Tensor, txt_len: int, cu_seqlens_q
hyvideo/modules/models.py:326
Methodforward
( self, x: torch.Tensor, t: torch.Tensor, # Should be in range(0, 1000). text
hyvideo/modules/models.py:595
Methodforward
( self, x: torch.Tensor, c: torch.Tensor, # timestep_aware_representations + context_
hyvideo/modules/token_refiner.py:77
Methodforward
( self, x: torch.Tensor, c: torch.LongTensor, mask: Optional[torch.Tensor] = N
hyvideo/modules/token_refiner.py:137
Methodforward
( self, x: torch.Tensor, t: torch.LongTensor, mask: Optional[torch.LongTensor]
hyvideo/modules/token_refiner.py:214
Methodforward
(self, x)
hyvideo/modules/mlp_layers.py:52
Methodforward
(self, x: torch.Tensor)
hyvideo/modules/mlp_layers.py:72
Methodforward
(self, x, c)
hyvideo/modules/mlp_layers.py:114
Methodforward
(self, x: torch.Tensor)
hyvideo/modules/modulate_layers.py:27
Methodfuse_qkv_projections
Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value) are fused. For cross-a
hyvideo/vae/autoencoder_kl_causal_3d.py:568
Methodget_guidance_scale_embedding
See https://github.com/google-research/vdm/blob/dc27b98a554f65cdc654b800da5aa1846545d41b/model_vdm.py#L298 Args: w (`tor
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:597
Methodguidance_rescale
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:635
Methodguidance_scale
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:631
Methodinterrupt
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:659
Methodkl
(self, other: "DiagonalGaussianDistribution" = None)
hyvideo/vae/vae.py:324
Functionnew_forward
( self, x: torch.Tensor, t: torch.Tensor, # Should be in range(0, 1000). text
hyvideo/inference.py:45
Methodnll
(self, sample: torch.Tensor, dims: Tuple[int, ...] = [1, 2, 3])
hyvideo/vae/vae.py:344
Methodnum_timesteps
(self)
hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py:655
Methodparams_count
(self)
hyvideo/modules/models.py:712
Functionparse
(x)
hyvideo/utils/helpers.py:7
Methodparse_size
(size)
hyvideo/inference.py:357
Functionsafe_dir
Create a directory (or the parent directory of a file) if it does not exist. Args: path (str or Path): Path to the directory. R
hyvideo/utils/file_utils.py:18
Functionsafe_file
Create the parent directory of a file if it does not exist. Args: path (str or Path): Path to the file. Returns: path (
hyvideo/utils/file_utils.py:33
Methodset_begin_index
Sets the begin index for the scheduler. This function should be run from pipeline before the inference. Args: begin_inde
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:112
Methodset_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
hyvideo/vae/autoencoder_kl_causal_3d.py:244
Methodstep_index
The index counter for current timestep. It will increase 1 after each scheduler step.
hyvideo/diffusion/schedulers/scheduling_flow_match_discrete.py:98
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
hyvideo/vae/autoencoder_kl_causal_3d.py:592
← previous101–197 of 197, ranked by callers