[causal attn mask] replaced for loop over tensor with pytorch tensor ops#208
Open
dabeschte wants to merge 2 commits intoTencent-Hunyuan:mainfrom
Open
[causal attn mask] replaced for loop over tensor with pytorch tensor ops#208dabeschte wants to merge 2 commits intoTencent-Hunyuan:mainfrom
dabeschte wants to merge 2 commits intoTencent-Hunyuan:mainfrom