↓ 1 callersFunctionteacache_forward Args: x: (B, C, T, H, W) tensor of spatial-temp inputs timesteps: (B, ) tensor of timesteps crossattn_emb: (B, N, D) tens
TeaCache4Cosmos/teacache_sample_video_i2v.py:31
↓ 1 callersMethodtemporal_attention(
self,
attn,
hidden_states,
residual,
batch_size,
batchsize,
videosys/models/modules/attentions.py:601
↓ 1 callersMethodtorch_impl(self, q, k, v, mask, B, N, C)
videosys/models/modules/attentions.py:155
Method__call__ Args: clip (torch.tensor, dtype=torch.uint8): Size is (T, C, H, W) Return: clip (torch.tensor, dtype=torch.fl
videosys/pipelines/open_sora/data_process.py:671