Functions — github.com/ali-vilab/TeaCache

CodeHub MCPcopy Create free account

hub / github.com/ali-vilab/TeaCache / functions

Functions1,040 in github.com/ali-vilab/TeaCache

⨍ Functions 1,040◇ Types & classes 227

(self, *args, **kwargs)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:157

(self, parameters, deterministic=False)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:200

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:241

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:278

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int]] = (3, 3),

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:319

(self, in_channels, out_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:345

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:365

(self, in_channels, out_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:419

(self, in_channels, out_channels, undown=False)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:431

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:468

( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:510

( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:630

( self, hidden_size: int = 128, z_channels: int = 4, hidden_size_mult: Tuple[i

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:743

(self, model_path, subfolder=None, cache_dir=None, use_ema=False, **kwargs)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:1113

( self, in_channels: int, out_channels: int, kernel_size: Union[int, Tuple[int

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:72

( self, f_channels: int, zq_channels: int, groups: int = 32, )

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:155

( self, in_channels: int, out_channels: Optional[int] = None, dropout: float =

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:209

( self, in_channels: int, out_channels: int, temb_channels: int, dropo

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:334

( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:436

( self, in_channels: int = 3, out_channels: int = 16, down_block_types: Tuple[

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:622

( self, in_channels: int = 16, out_channels: int = 3, up_block_types: Tuple[st

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:756

( self, in_channels: int = 3, out_channels: int = 3, down_block_types: Tuple[s

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:909

(self, parameters, deterministic=False)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:46

(self, *args, **kwargs)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:93

( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:135

( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:252

( self, lr: float = 1e-5, hidden_size: int = 128, z_channels: int = 4,

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:359

(self, model_path, subfolder=None, cache_dir=None, **kwargs)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:801

(self, dim, heads=4, dim_head=32)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:871

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:892

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:899

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:941

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:992

(self, in_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1031

( self, in_channels: int, out_channels: int, kernel_size: Union[int, Tuple[int

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1081

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int, int, int]], init_method="random", **kwar

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1115

(self, num_channels, num_groups=32, eps=1e-6, *args, **kwargs)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1168

(self, num_features, logdet=False, affine=True, allow_reverse_init=False)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1181

(self, n_codes, embedding_dim)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1284

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1379

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1416

(self, in_channels, out_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1451

(self, in_channels, out_channels)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1466

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int]] = (3, 3),

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1485

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int]] = (3, 3),

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1508

(self, chan_in, chan_out, kernel_size: int = 3)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1531

(self, chan_in, chan_out)

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1543

( self, in_channels, out_channels, kernel_size: int = 3, mix_factor: f

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1555

( self, in_channels, out_channels, kernel_size: int = 3, mix_factor: f

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1576

( self, in_channels, out_channels, kernel_size: int = 3, mix_factor: f

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1597

( self, in_channels, out_channels, kernel_size: int = 3, mix_factor: f

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1620

( self, parameters, deterministic=False, )

videosys/models/autoencoders/autoencoder_kl_open_sora.py:22

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int, int, int]],

videosys/models/autoencoders/autoencoder_kl_open_sora.py:90

( self, in_channels, # SCH: added filters, conv_fn, activation_fn=nn.

videosys/models/autoencoders/autoencoder_kl_open_sora.py:128

( self, in_out_channels=4, latent_embed_dim=512, filters=128, num_res_

videosys/models/autoencoders/autoencoder_kl_open_sora.py:278

( self, in_out_channels=4, latent_embed_dim=4, embed_dim=4, filters=12

videosys/models/autoencoders/autoencoder_kl_open_sora.py:380

( self, from_pretrained=None, micro_batch_size=None, cache_dir=None, local_files_only=False, subfolder

videosys/models/autoencoders/autoencoder_kl_open_sora.py:489

(self, from_pretrained=None, cache_dir=None, local_files_only=False)

videosys/models/autoencoders/autoencoder_kl_open_sora.py:559

( self, vae_2d=None, vae_temporal=None, from_pretrained=None, freeze_v

videosys/models/autoencoders/autoencoder_kl_open_sora.py:598

(self, config: VideoAutoencoderPipelineConfig)

videosys/models/autoencoders/autoencoder_kl_open_sora.py:624

videosys/schedulers/scheduling_dpm_cogvideox.py:482

videosys/schedulers/scheduling_ddim_cogvideox.py:442

videosys/pipelines/open_sora/data_process.py:632

videosys/pipelines/open_sora/data_process.py:680

videosys/pipelines/open_sora/data_process.py:695

Method_basic_init

videosys/models/transformers/open_sora_transformer_3d.py:460

Method_driver_execute_model

(self, *args, **kwargs)

videosys/core/engine.py:97

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:488

Method_get_signature_keys

videosys/core/pipeline.py:34

Method_get_variance

(self, timestep, prev_timestep)

videosys/schedulers/scheduling_dpm_cogvideox.py:228

Method_get_variance

(self, timestep, prev_timestep)

videosys/schedulers/scheduling_ddim_cogvideox.py:227

Function_run_worker_process

Worker process event loop

videosys/core/mp_utils.py:181

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/open_sora_plan_v120_transformer_3d.py:1730

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/vchitect_transformer_3d.py:479

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/cogvideox_transformer_3d.py:476

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/latte_transformer_3d.py:1141

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:2378

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/transformers/open_sora_plan_transformer_3d.py:2373

Method_set_gradient_checkpointing

(self, module, value=False)

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:1004

Methodadd_noise

( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.

videosys/schedulers/scheduling_dpm_cogvideox.py:436

Methodadd_noise

( self, original_samples: torch.Tensor, noise: torch.Tensor, timesteps: torch.

videosys/schedulers/scheduling_ddim_cogvideox.py:396

Functionadd_watermark

( input_video_path, watermark_image_path="./assets/images/watermark/watermark.png", output_video_path=None

videosys/pipelines/open_sora/pipeline_open_sora.py:962

Functionall_exists

videosys/utils/utils.py:81

Methodattn_processors

r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with

videosys/models/transformers/vchitect_transformer_3d.py:381

(ctx: Any, *grad_outputs)

videosys/core/comm.py:77

(ctx: Any, *grad_outputs)

videosys/core/comm.py:150

(ctx: Any, *grad_outputs)

videosys/core/comm.py:194

(ctx, *grad_output)

videosys/core/comm.py:235

(ctx, grad_output)

videosys/core/comm.py:318

(ctx, grad_output)

videosys/core/comm.py:350

Methodbatch_to_head_dim

r""" Reshape the tensor from `[batch_size, seq_len, dim]` to `[batch_size // heads, seq_len, dim * heads]`. `heads` is the number of h

videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:938

Methodbatch_to_head_dim

r""" Reshape the tensor from `[batch_size, seq_len, dim]` to `[batch_size // heads, seq_len, dim * heads]`. `heads` is the number of h

videosys/models/transformers/open_sora_plan_transformer_3d.py:951

Methodclip_skip

videosys/pipelines/vchitect/pipeline_vchitect.py:688

videosys/core/mp_utils.py:143

Methodconfig_sanity_check

videosys/core/shardformer/t5/policy.py:7

Methodconfigure_optimizers

videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:592

Methodconfigure_optimizers

videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:592

Methodcreate_custom_forward

(module, return_dict=None)

videosys/models/transformers/open_sora_plan_v120_transformer_3d.py:1905

Methodcreate_custom_forward

videosys/models/transformers/cogvideox_transformer_3d.py:540

Methodcreate_custom_forward

videosys/models/autoencoders/autoencoder_kl_cogvideox.py:390

← previous next →601–700 of 1,040, ranked by callers