MCPcopy Create free account

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

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

⨍ Functions 283◇ Types & classes 50
Methodforward
(self, inputs: torch.Tensor)
videosys/modules/upsampling.py:39
Methodforward
r""" Args: text_embeds (`torch.Tensor`): Input text embeddings. Expected shape: (batch_size, seq_length, embedding
videosys/modules/embeddings.py:28
Methodforward
Forward function.
videosys/modules/embeddings.py:82
Methodforward
(self, t, dtype)
videosys/modules/embeddings.py:138
Methodforward
(self, s, bs)
videosys/modules/embeddings.py:161
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb: torch.Tensor )
videosys/modules/normalization.py:22
Methodforward
( self, x: torch.Tensor, timestep: Optional[torch.Tensor] = None, temb: Optional[torch.Tensor] = None
videosys/modules/normalization.py:67
Methodforward
r""" Args: text_embeds (`torch.Tensor`): Input text embeddings. Expected shape: (batch_size, seq_length, embedding
cogvideox_fun/transformer_3d.py:227
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb
cogvideox_fun/transformer_3d.py:330
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, time
cogvideox_fun/transformer_3d.py:667
Methodforward
(self, x, *args, **kwargs)
cogvideox_fun/lora_utils.py:81
Methodforward
(self, input: torch.Tensor)
cogvideox_fun/autoencoder_magvit.py:43
Methodforward
(self, inputs: torch.Tensor)
cogvideox_fun/autoencoder_magvit.py:133
Methodforward
(self, f: torch.Tensor, zq: torch.Tensor)
cogvideox_fun/autoencoder_magvit.py:175
Methodforward
( self, inputs: torch.Tensor, temb: Optional[torch.Tensor] = None, zq: Optiona
cogvideox_fun/autoencoder_magvit.py:277
Methodforward
( self, hidden_states: torch.Tensor, temb: Optional[torch.Tensor] = None, zq:
cogvideox_fun/autoencoder_magvit.py:390
Methodforward
( self, hidden_states: torch.Tensor, temb: Optional[torch.Tensor] = None, zq:
cogvideox_fun/autoencoder_magvit.py:478
Methodforward
r"""The forward method of the `CogVideoXEncoder3D` class.
cogvideox_fun/autoencoder_magvit.py:708
Methodforward
r"""The forward method of the `CogVideoXDecoder3D` class.
cogvideox_fun/autoencoder_magvit.py:849
Methodforward
( self, sample: torch.Tensor, sample_posterior: bool = False, return_dict: boo
cogvideox_fun/autoencoder_magvit.py:1280
Methodforward
r""" Args: text_embeds (`torch.Tensor`): Input text embeddings. Expected shape: (batch_size, seq_length, embedding
cogvideox_fun/fun_pab_transformer_3d.py:228
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb
cogvideox_fun/fun_pab_transformer_3d.py:340
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, time
cogvideox_fun/fun_pab_transformer_3d.py:579
Functionfp8_linear_forward
(cls, x)
mz_cogvideox_core.py:193
Methodfrom_pretrained_2d
(cls, pretrained_model_path, subfolder=None, transformer_additional_kwargs={})
cogvideox_fun/transformer_3d.py:818
Methodfrom_pretrained_2d
(cls, pretrained_model_path, subfolder=None, transformer_additional_kwargs={})
cogvideox_fun/fun_pab_transformer_3d.py:680
Methodfuse_qkv_projections
r"""Enables fused QKV projections.
pipeline_cogvideox.py:299
Methodfuse_qkv_projections
Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value) are fused. For cross-att
custom_cogvideox_transformer_3d.py:585
Methodfuse_qkv_projections
Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value) are fused. For cross-att
cogvideox_fun/transformer_3d.py:628
Methodfuse_qkv_projections
r"""Enables fused QKV projections.
cogvideox_fun/pipeline_cogvideox_control.py:414
Methodfuse_qkv_projections
r"""Enables fused QKV projections.
cogvideox_fun/pipeline_cogvideox_inpaint.py:498
Methodfuse_qkv_projections
Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value) are fused. For cross-att
cogvideox_fun/fun_pab_transformer_3d.py:540
Functionget_closest_ratio
(height: float, width: float, ratios: dict = ASPECT_RATIO_512)
cogvideox_fun/utils.py:52
Functionget_image_to_video_latent
(validation_image_start, validation_image_end, video_length, sample_size)
cogvideox_fun/utils.py:66
Functionget_missing_indexes
(windows: list[list[int]], num_frames: int)
cogvideox_fun/context.py:37
Functionget_mlp_output
(skip_range, timestep, block_idx: int, is_temporal=False)
videosys/core/pab_mgr.py:231
Methodget_spatial_mlp_outputs
(self)
videosys/core/pab_mgr.py:176
Methodget_temporal_mlp_outputs
(self)
videosys/core/pab_mgr.py:179
Functionget_total_steps
( scheduler, timesteps: List[int], num_steps: Optional[int] = None, num_frames: int = ...,
cogvideox_fun/context.py:160
Methodget_trainable_params
(self)
cogvideox_fun/lora_utils.py:313
Functionget_video_to_video_latent
(input_video_path, video_length, sample_size, validation_video_mask=None)
cogvideox_fun/utils.py:162
Functionget_width_and_height_from_image_and_base_resolution
(image, base_resolution)
cogvideox_fun/utils.py:58
Methodguidance_scale
(self)
cogvideox_fun/pipeline_cogvideox_control.py:471
Methodguidance_scale
(self)
cogvideox_fun/pipeline_cogvideox_inpaint.py:555
Functionif_broadcast_cross
(timestep: int, count: int)
videosys/core/pab_mgr.py:203
Functionif_broadcast_mlp
(timestep: int, count: int, block_idx: int, all_timesteps, is_temporal=False)
videosys/core/pab_mgr.py:221
Functionif_broadcast_temporal
(timestep: int, count: int)
videosys/core/pab_mgr.py:209
Methodinterrupt
(self)
pipeline_cogvideox.py:363
Methodinterrupt
(self)
cogvideox_fun/pipeline_cogvideox_control.py:479
Methodinterrupt
(self)
cogvideox_fun/pipeline_cogvideox_inpaint.py:563
Functionload_lora_into_transformer
(state_dict, transformer, adapter_name=None)
cogvideox_fun/lora_utils.py:479
Methodload_weights
(self, file)
cogvideox_fun/lora_utils.py:276
Functionmerge_lora
(transformer, lora_path, multiplier, device='cpu', dtype=torch.float32, state_dict=None)
cogvideox_fun/lora_utils.py:369
Methodnum_timesteps
(self)
pipeline_cogvideox.py:359
Methodnum_timesteps
(self)
cogvideox_fun/pipeline_cogvideox_control.py:475
Methodnum_timesteps
(self)
cogvideox_fun/pipeline_cogvideox_inpaint.py:559
Functionpost_to_cpu
(module, inp, out)
mz_dyn_cpu_offload.py:69
Functionpre_blocks_forward_hook
(module, inp)
mz_dyn_cpu_offload.py:80
Functionpre_to_cuda
(module, inp)
mz_dyn_cpu_offload.py:63
Methodprepare_control_latents
( self, mask, masked_image, batch_size, height, width, dtype, device, generator, do_classifier_free_gu
cogvideox_fun/pipeline_cogvideox_control.py:289
Methodprepare_optimizer_params
(self, text_encoder_lr, unet_lr, default_lr)
cogvideox_fun/lora_utils.py:286
Functionquantize_load_state_dict
(model, state_dict, device="cpu", cast_dtype=torch.float16)
mz_gguf_loader.py:19
Functionregister_hooks
(cls, l)
mz_dyn_cpu_offload.py:107
Functionretrieve_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_inpaint.py:87
Functionsave_mlp_output
(timestep: int, block_idx: int, ff_output, is_temporal=False)
videosys/core/pab_mgr.py:227
Methodsave_weights
(self, file, dtype, metadata)
cogvideox_fun/lora_utils.py:316
Methodset_eval_and_device
(device: torch.device, *modules)
videosys/core/pipeline.py:12
Methodset_multiplier
(self, multiplier)
cogvideox_fun/lora_utils.py:271
Functionstatic_standard
( step: int = ..., num_steps: Optional[int] = None, num_frames: int = ..., context_size: Optio
cogvideox_fun/context.py:123
Functiontiled_encode
r"""Encode a batch of images using a tiled encoder. When this option is enabled, the VAE will split the input tensor into tiles to compute encodin
mz_enable_vae_encode_tiling.py:38
Functionto
(self, *args, **kwargs)
mz_dyn_cpu_offload.py:46
Functionto_pil
(image)
cogvideox_fun/utils.py:15
Methodunfuse_qkv_projections
r"""Disable QKV projection fusion if enabled.
pipeline_cogvideox.py:304
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
custom_cogvideox_transformer_3d.py:611
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
cogvideox_fun/transformer_3d.py:654
Methodunfuse_qkv_projections
r"""Disable QKV projection fusion if enabled.
cogvideox_fun/pipeline_cogvideox_control.py:419
Methodunfuse_qkv_projections
r"""Disable QKV projection fusion if enabled.
cogvideox_fun/pipeline_cogvideox_inpaint.py:503
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
cogvideox_fun/fun_pab_transformer_3d.py:566
Functionuniform_looped
( step: int = ..., num_steps: Optional[int] = None, num_frames: int = ..., context_size: Optio
cogvideox_fun/context.py:47
Functionuniform_standard
( step: int = ..., num_steps: Optional[int] = None, num_frames: int = ..., context_size: Optio
cogvideox_fun/context.py:72
Functionunmerge_lora
Unmerge state_dict in LoRANetwork from the pipeline in diffusers.
cogvideox_fun/lora_utils.py:427
Functionunregister_hooks
(self)
mz_dyn_cpu_offload.py:133
Functionupdate_steps
(steps: int)
videosys/core/pab_mgr.py:198
← previous201–283 of 283, ranked by callers