Replace VMap that Fails Torch.Export#42721

Closed

YifanShenSZ wants to merge 3 commits intohuggingface:v4.57-releasefrom

YifanShenSZ:v4.57.3_torch-export

Commits on Dec 8, 2025

torch.export does not support such vmap usage, so use simple causal mask construction
YifanShenSZ
committed
models such as GPT-OSS require 4D mask
YifanShenSZ
committed
use -1 in reshape to dodge Llama4 dynamic-shape torch.export failure
YifanShenSZ
committed