Skip to content

[MPS] CrossAttention: (partial) fix for NaN attention_scores on PyTorch nightly#2643

Open
Birch-san wants to merge 1 commit into
huggingface:mainfrom
Birch-san:mps-attn-bias-alloc
Open

[MPS] CrossAttention: (partial) fix for NaN attention_scores on PyTorch nightly#2643
Birch-san wants to merge 1 commit into
huggingface:mainfrom
Birch-san:mps-attn-bias-alloc

for the (unused) attention bias which the baddbmm API requires us to …

cdf071a
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs