return total_second_length * 60 * (1.5 if use_teacache else 2.5) * (1 + ((steps - 25) / 100)) 14e3ba4 verified Fabrice-TIERCELIN commited on Jun 21, 2025
self_attn_mask_1 = attention_mask.view(batch_size, 1, 1, seq_len).expand(-1, -1, seq_len, -1) 447d433 verified Fabrice-TIERCELIN commited on Jun 21, 2025
self_attn_mask_1 = attention_mask.view(batch_size, 1, 1, seq_len).expand(1, 1, seq_len, 1) 7542c74 verified Fabrice-TIERCELIN commited on Jun 21, 2025