Skip to content

Replace VMap that Fails Torch.Export - #42721

Closed
YifanShenSZ wants to merge 3 commits into
huggingface:v4.57-releasefrom
YifanShenSZ:v4.57.3_torch-export
Closed

Replace VMap that Fails Torch.Export#42721
YifanShenSZ wants to merge 3 commits into
huggingface:v4.57-releasefrom
YifanShenSZ:v4.57.3_torch-export

use -1 in reshape to dodge Llama4 dynamic-shape torch.export failure

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