Skip to content

Cast attention inputs to the autocast dtype in dispatch_attention_fn#14160

Draft
danielxmed wants to merge 5 commits into
huggingface:mainfrom
danielxmed:fix/issue-14104-autocast-attention-dtype
Draft

Cast attention inputs to the autocast dtype in dispatch_attention_fn#14160
danielxmed wants to merge 5 commits into
huggingface:mainfrom
danielxmed:fix/issue-14104-autocast-attention-dtype

Gate the dispatcher autocast tests behind require_torch_accelerator a…

e864233
Select commit
Loading
Failed to load commit list.
Sign in for the full log view