↓ 7 callersFunctionflash_attention q: [B, Lq, Nq, C1]. k: [B, Lk, Nk, C1]. v: [B, Lk, Nk, C2]. Nq must be divisible by Nk. q_lens
dreamidv_wan/modules/attention.py:24
↓ 6 callersFunctionflash_attention q: [B, Lq, Nq, C1]. k: [B, Lk, Nk, C1]. v: [B, Lk, Nk, C2]. Nq must be divisible by Nk. q_lens
dreamidv_wan_faster/modules/attention.py:24
↓ 3 callersFunctionrope_apply x: [B, L, N, C]. grid_sizes: [B, 3]. freqs: [M, C // 2].
dreamidv_wan/distributed/xdit_context_parallel.py:33
↓ 2 callersMethodextend(self, prompt, system_prompt, seed=-1, *args, **kwargs)
dreamidv_wan_faster/utils/prompt_extend.py:128
↓ 2 callersFunctionrope_apply x: [B, L, N, C]. grid_sizes: [B, 3]. freqs: [M, C // 2].
dreamidv_wan_faster/distributed/xdit_context_parallel.py:33
↓ 2 callersFunctionsave_visualization_video(landmarks_sequence, output_filename, frame_size, fps=30, mode='points', optional_face_oval_indices=None)
express_adaption/align_refvideo.py:61