MCPcopy Create free account

hub / github.com/MinusZoneAI/ComfyUI-CogVideoX-MZ / functions

Functions283 in github.com/MinusZoneAI/ComfyUI-CogVideoX-MZ

↓ 16 callersFunctionapply_rotary_emb
Apply rotary embeddings to input tensors using the given frequency tensor. This function applies rotary embeddings to the given query or key
videosys/modules/embeddings.py:263
↓ 11 callersFunctionenable_pab
()
videosys/core/pab_mgr.py:188
↓ 8 callersMethod__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
cogvideox_fun/autoencoder_magvit.py:533
↓ 6 callersMethodencode
Encode a batch of images into latents. Args: x (`torch.Tensor`): Input batch of images. return_dict (`bool`,
cogvideox_fun/autoencoder_magvit.py:1085
↓ 5 callersMethodload
(self, **kwargs)
__init__.py:48
↓ 4 callersMethod_clear_fake_context_parallel_cache
(self)
cogvideox_fun/autoencoder_magvit.py:129
↓ 4 callersMethod_prepare_rotary_positional_embeddings
( self, height: int, width: int, num_frames: int, device: torch.device
pipeline_cogvideox.py:312
↓ 4 callersFunctionordered_halving
(val)
cogvideox_fun/context.py:5
↓ 4 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
cogvideox_fun/transformer_3d.py:593
↓ 3 callersMethod__init__
(self, hidden_size, frequency_embedding_size=256)
videosys/modules/embeddings.py:151
↓ 3 callersMethod_clear_fake_context_parallel_cache
(self)
cogvideox_fun/autoencoder_magvit.py:1021
↓ 3 callersMethod_prepare_rotary_positional_embeddings
( self, height: int, width: int, num_frames: int, device: torch.device
cogvideox_fun/pipeline_cogvideox_control.py:427
↓ 3 callersMethod_prepare_rotary_positional_embeddings
( self, height: int, width: int, num_frames: int, device: torch.device
cogvideox_fun/pipeline_cogvideox_inpaint.py:511
↓ 3 callersMethoddecode
Decode a batch of images. Args: z (`torch.Tensor`): Input batch of latent vectors. return_dict (`bool`, *opt
cogvideox_fun/autoencoder_magvit.py:1160
↓ 3 callersFunctionfft
(tensor)
custom_cogvideox_transformer_3d.py:46
↓ 3 callersFunctionfft
(tensor)
cogvideox_fun/transformer_3d.py:46
↓ 3 callersFunctionget_3d_rotary_pos_embed
RoPE for video tokens with 3D structure. Args: embed_dim: (`int`): The embedding dimension size, corresponding to hidden_size_he
videosys/modules/embeddings.py:179
↓ 3 callersFunctionget_context_scheduler
(name: str)
cogvideox_fun/context.py:149
↓ 3 callersFunctionset_pab_manager
(config: PABConfig)
videosys/core/pab_mgr.py:183
↓ 2 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
cogvideox_fun/transformer_3d.py:285
↓ 2 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
cogvideox_fun/fun_pab_transformer_3d.py:286
↓ 2 callersMethod_decode
(self, z: torch.Tensor, return_dict: bool = True)
cogvideox_fun/autoencoder_magvit.py:1125
↓ 2 callersMethodblend_h
(self, a: torch.Tensor, b: torch.Tensor, blend_extent: int)
cogvideox_fun/autoencoder_magvit.py:1192
↓ 2 callersMethodblend_v
(self, a: torch.Tensor, b: torch.Tensor, blend_extent: int)
cogvideox_fun/autoencoder_magvit.py:1184
↓ 2 callersMethodcompress_time
(self, x, num_frames)
cogvideo_controlnet.py:126
↓ 2 callersFunctionconvert_fp8_linear
(module, dtype, cast_dtype)
mz_cogvideox_core.py:238
↓ 2 callersMethodforward
r"""Forward method of the `CogVideoXUpBlock3D` class.
cogvideox_fun/autoencoder_magvit.py:582
↓ 2 callersFunctionif_broadcast_spatial
(timestep: int, count: int, block_idx: int)
videosys/core/pab_mgr.py:215
↓ 2 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
custom_cogvideox_transformer_3d.py:550
↓ 2 callersMethodtimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These ma
videosys/modules/embeddings.py:119
↓ 2 callersFunctiontransformer_blocks_to_cpu
(m, layer_start=0, layer_size=-1)
mz_dyn_cpu_offload.py:11
↓ 1 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
custom_cogvideox_transformer_3d.py:247
↓ 1 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
videosys/cogvideox_transformer_3d.py:236
↓ 1 callersMethod__init__
( self, embedding_dim: int, num_embeddings: Optional[int] = None, output_dim:
videosys/modules/normalization.py:44
↓ 1 callersMethod__init__
if alpha == 0 or None, alpha is rank (no scaling).
cogvideox_fun/lora_utils.py:27
↓ 1 callersMethod_gaussian_weights
(self, t_tile_length, t_batch_size)
pipeline_cogvideox.py:289
↓ 1 callersMethod_gaussian_weights
(self, t_tile_length, t_batch_size)
cogvideox_fun/pipeline_cogvideox_control.py:352
↓ 1 callersMethod_gaussian_weights
(self, t_tile_length, t_batch_size)
cogvideox_fun/pipeline_cogvideox_inpaint.py:436
↓ 1 callersMethod_is_t_in_skip_config
(all_timesteps, timestep, config)
videosys/core/pab_mgr.py:94
↓ 1 callersFunctionadd_noise_to_reference_video
(image, ratio=None)
cogvideox_fun/pipeline_cogvideox_inpaint.py:182
↓ 1 callersFunctionaddnet_hash_legacy
Old model hash used by sd-webui-additional-networks for .safetensors format files
cogvideox_fun/lora_utils.py:115
↓ 1 callersFunctionaddnet_hash_safetensors
New model hash used by sd-webui-additional-networks for .safetensors format files
cogvideox_fun/lora_utils.py:124
↓ 1 callersMethodapply_to
(self)
cogvideox_fun/lora_utils.py:76
↓ 1 callersFunctionbroadcast
(tensors, dim=-1)
videosys/modules/embeddings.py:229
↓ 1 callersMethodcheck_inputs
( self, height, width, prompt_embeds=None, negative_prompt_embeds=None
pipeline_cogvideox.py:261
↓ 1 callersMethodcheck_inputs
( self, prompt, height, width, negative_prompt, callback_on_st
cogvideox_fun/pipeline_cogvideox_control.py:363
↓ 1 callersMethodcheck_inputs
( self, prompt, height, width, negative_prompt, callback_on_st
cogvideox_fun/pipeline_cogvideox_inpaint.py:447
↓ 1 callersFunctiondequantize_blocks_Q4_0
(data, dtype=torch.float16)
mz_gguf_loader.py:160
↓ 1 callersFunctiondoes_window_roll_over
(window: list[int], num_frames: int)
cogvideox_fun/context.py:12
↓ 1 callersFunctionenable_vae_encode_tiling
(vae)
mz_enable_vae_encode_tiling.py:189
↓ 1 callersMethodfake_context_parallel_forward
(self, inputs: torch.Tensor)
cogvideox_fun/autoencoder_magvit.py:120
↓ 1 callersMethodfrom_linear
( cls, linear, device="cpu", qtype="Q4_0", cast_dtype=torch.float16, )
mz_gguf_loader.py:93
↓ 1 callersFunctiongen_fp8_linear_forward
(cast_dtype)
mz_cogvideox_core.py:192
↓ 1 callersMethodgenerate
(self, *args, **kwargs)
videosys/core/pipeline.py:18
↓ 1 callersFunctiongenerate_transformer_blocks_forward_hook
(cls, layer_start, layer_size)
mz_dyn_cpu_offload.py:79
↓ 1 callersMethodget_mlp_output
(self, skip_range, timestep, block_idx, is_temporal=False)
videosys/core/pab_mgr.py:147
↓ 1 callersFunctionget_resize_crop_region_for_grid
(src, tgt_width, tgt_height)
pipeline_cogvideox.py:40
↓ 1 callersFunctionget_resize_crop_region_for_grid
(src, tgt_width, tgt_height)
cogvideox_fun/pipeline_cogvideox_control.py:68
↓ 1 callersFunctionget_resize_crop_region_for_grid
(src, tgt_width, tgt_height)
cogvideox_fun/pipeline_cogvideox_inpaint.py:68
↓ 1 callersMethodget_timesteps
(self, num_inference_steps, strength, device)
pipeline_cogvideox.py:278
↓ 1 callersMethodget_timesteps
(self, num_inference_steps, strength, device)
cogvideox_fun/pipeline_cogvideox_control.py:483
↓ 1 callersMethodget_timesteps
(self, num_inference_steps, strength, device)
cogvideox_fun/pipeline_cogvideox_inpaint.py:567
↓ 1 callersMethodif_broadcast_cross
(self, timestep: int, count: int)
videosys/core/pab_mgr.py:54
↓ 1 callersMethodif_broadcast_spatial
(self, timestep: int, count: int, block_idx: int)
videosys/core/pab_mgr.py:80
↓ 1 callersMethodif_broadcast_temporal
(self, timestep: int, count: int)
videosys/core/pab_mgr.py:67
↓ 1 callersMethodif_skip_mlp
(self, timestep: int, count: int, block_idx: int, all_timesteps, is_temporal=False)
videosys/core/pab_mgr.py:108
↓ 1 callersFunctionnumpy2pil
(image)
cogvideox_fun/utils.py:12
↓ 1 callersFunctionprecalculate_safetensors_hashes
Precalculate the model hashes needed by sd-webui-additional-networks to save time on indexing the model later.
cogvideox_fun/lora_utils.py:141
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
pipeline_cogvideox.py:243
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
cogvideox_fun/pipeline_cogvideox_control.py:335
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
cogvideox_fun/pipeline_cogvideox_inpaint.py:419
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, timeste
pipeline_cogvideox.py:169
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, timeste
cogvideox_fun/pipeline_cogvideox_control.py:216
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, height, width, vide
cogvideox_fun/pipeline_cogvideox_inpaint.py:264
↓ 1 callersMethodprepare_mask_latents
( self, mask, masked_image, batch_size, height, width, dtype, device, generator, do_classifier_free_gu
cogvideox_fun/pipeline_cogvideox_inpaint.py:371
↓ 1 callersFunctionquant_shape_from_byte_shape
(shape, qtype)
mz_gguf_loader.py:146
↓ 1 callersFunctionquant_shape_to_byte_shape
(shape, qtype)
mz_gguf_loader.py:137
↓ 1 callersFunctionremove_specific_blocks
(model, block_indices_to_remove)
mz_cogvideox_core.py:438
↓ 1 callersFunctionresize_mask
(mask, latent, process_first_frame_only=True)
cogvideox_fun/pipeline_cogvideox_inpaint.py:146
↓ 1 callersFunctionretrieve_timesteps
Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwargs
pipeline_cogvideox.py:58
↓ 1 callersFunctionretrieve_timesteps
Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwargs
cogvideox_fun/pipeline_cogvideox_control.py:87
↓ 1 callersMethodsave_skip_output
(self, timestep, block_idx, ff_output, is_temporal=False)
videosys/core/pab_mgr.py:141
↓ 1 callersFunctionset_op_by_name
(layer, name, new_module)
mz_gguf_loader.py:41
↓ 1 callersFunctionshift_window_to_end
(window: list[int], num_frames: int)
cogvideox_fun/context.py:28
↓ 1 callersFunctionshift_window_to_start
(window: list[int], num_frames: int)
cogvideox_fun/context.py:21
↓ 1 callersFunctionsplit_block_dims
(blocks, *args)
mz_gguf_loader.py:131
↓ 1 callersFunctiontensor2pil
(image)
cogvideox_fun/utils.py:9
↓ 1 callersMethodtiled_decode
r""" Decode a batch of images using a tiled decoder. Args: z (`torch.Tensor`): Input batch of latent vectors.
cogvideox_fun/autoencoder_magvit.py:1200
↓ 1 callersFunctiontransformer_blocks_to_cuda
(m, layer_start=0, layer_size=-1)
mz_dyn_cpu_offload.py:21
MethodINPUT_TYPES
(s)
__init__.py:24
FunctionMZ_CogVideoXLoader_call
(args={})
mz_cogvideox_core.py:248
Method__call__
Function invoked when calling the pipeline for generation. Args: height (`int`, *optional*, defaults to self.unet.config
pipeline_cogvideox.py:367
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
custom_cogvideox_transformer_3d.py:73
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
custom_cogvideox_transformer_3d.py:147
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
videosys/cogvideox_transformer_3d.py:46
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
videosys/cogvideox_transformer_3d.py:138
Method__call__
In diffusers, it is a convention to call the pipeline object. But in VideoSys, we will use the generate method for better prompt.
videosys/core/pipeline.py:21
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
cogvideox_fun/transformer_3d.py:73
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: t
cogvideox_fun/transformer_3d.py:147
Method__call__
Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
cogvideox_fun/pipeline_cogvideox_control.py:494
next →1–100 of 283, ranked by callers