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
Methodforward
(self, tokens, positions)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:247
Methodforward
(self, timestep, resolution, aspect_ratio, batch_size, hidden_dtype)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:324
Methodforward
(self, caption, force_drop_ids=None)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:354
Methodforward
(self, latent)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:402
Methodforward
r""" The forward method of the `Attention` class. Args: hidden_states (`torch.Tensor`): The hidden states
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:904
Methodforward
(self, x: torch.Tensor, objs: torch.Tensor)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:1299
Methodforward
(self, hidden_states: torch.Tensor, scale: float = 1.0)
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:1359
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:1537
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:1910
Methodforward
( self, timestep: torch.Tensor, added_cond_kwargs: Dict[str, torch.Tensor] = None,
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:2095
Methodforward
The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix
videosys/models/transformers/open_sora_plan_v110_transformer_3d.py:2406
Methodforward
(self, tokens, positions)
videosys/models/transformers/open_sora_plan_transformer_3d.py:203
Methodforward
input: * tokens: batch_size x nheads x ntokens x dim * positions: batch_size x ntokens (t position of each token)
videosys/models/transformers/open_sora_plan_transformer_3d.py:242
Methodforward
(self, tokens, positions)
videosys/models/transformers/open_sora_plan_transformer_3d.py:260
Methodforward
(self, timestep, resolution, aspect_ratio, batch_size, hidden_dtype)
videosys/models/transformers/open_sora_plan_transformer_3d.py:337
Methodforward
(self, caption, force_drop_ids=None)
videosys/models/transformers/open_sora_plan_transformer_3d.py:367
Methodforward
(self, latent)
videosys/models/transformers/open_sora_plan_transformer_3d.py:415
Methodforward
r""" The forward method of the `Attention` class. Args: hidden_states (`torch.Tensor`): The hidden states
videosys/models/transformers/open_sora_plan_transformer_3d.py:917
Methodforward
(self, x: torch.Tensor, objs: torch.Tensor)
videosys/models/transformers/open_sora_plan_transformer_3d.py:1312
Methodforward
(self, hidden_states: torch.Tensor, scale: float = 1.0)
videosys/models/transformers/open_sora_plan_transformer_3d.py:1372
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
videosys/models/transformers/open_sora_plan_transformer_3d.py:1547
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
videosys/models/transformers/open_sora_plan_transformer_3d.py:1920
Methodforward
( self, timestep: torch.Tensor, added_cond_kwargs: Dict[str, torch.Tensor] = None,
videosys/models/transformers/open_sora_plan_transformer_3d.py:2107
Methodforward
The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix
videosys/models/transformers/open_sora_plan_transformer_3d.py:2401
Methodforward
(self, x, t, x_mask=None, t0=None, T=None, S=None)
videosys/models/transformers/open_sora_transformer_3d.py:74
Methodforward
( self, x, y, t, mask=None, # text mask x_mask=None, # tempo
videosys/models/transformers/open_sora_transformer_3d.py:154
Methodforward
( self, x, timestep, all_timesteps, y, mask=None, x_mask=None, fps=None, height=None, width=None, **kw
videosys/models/transformers/open_sora_transformer_3d.py:506
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:227
Methodforward
(self, z)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:335
Methodforward
(self, input, sample_posterior=True)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:493
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:825
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:881
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:912
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:954
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1006
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1044
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1113
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1163
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1175
Methodforward
(self, input, reverse=False)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1203
Methodforward
(self, z)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1320
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1400
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1436
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1461
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1477
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1503
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1524
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1539
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1549
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1571
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1590
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1615
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v110.py:1637
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:87
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:146
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:259
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:295
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:334
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:349
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:375
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:423
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:443
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:485
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:602
Methodforward
(self, z)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:713
Methodforward
(self, input, sample_posterior=True)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:884
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan_v120.py:1135
Methodforward
(self, input: torch.Tensor)
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:36
Methodforward
(self, inputs: torch.Tensor)
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:124
Methodforward
(self, f: torch.Tensor, zq: torch.Tensor)
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:166
Methodforward
( self, inputs: torch.Tensor, temb: Optional[torch.Tensor] = None, zq: Optiona
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:268
Methodforward
( self, hidden_states: torch.Tensor, temb: Optional[torch.Tensor] = None, zq:
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:381
Methodforward
( self, hidden_states: torch.Tensor, temb: Optional[torch.Tensor] = None, zq:
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:469
Methodforward
r"""The forward method of the `CogVideoXEncoder3D` class.
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:695
Methodforward
r"""The forward method of the `CogVideoXDecoder3D` class.
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:836
Methodforward
( self, sample: torch.Tensor, sample_posterior: bool = False, return_dict: boo
videosys/models/autoencoders/autoencoder_kl_cogvideox.py:1242
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:227
Methodforward
(self, z)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:335
Methodforward
(self, input, sample_posterior=True)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:493
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:878
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:909
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:951
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1003
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1041
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1110
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1160
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1172
Methodforward
(self, input, reverse=False)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1200
Methodforward
(self, z)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1317
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1397
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1433
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1458
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1474
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1500
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1521
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1536
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1546
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1568
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1587
Methodforward
(self, x)
videosys/models/autoencoders/autoencoder_kl_open_sora_plan.py:1612
← previousnext →801–900 of 1,040, ranked by callers