transformers
bugfix: torch.export failure caused by `_make_causal_mask`
#35291

Merged

bugfix: torch.export failure caused by `_make_causal_mask` #35291

ArthurZucker merged 2 commits into huggingface:main from jiwoong-choi:patch-1

bugfix: torch.export failure caused by `_make_causal_mask`

b2a00cf4

qubvel approved these changes on 2024-12-16

qubvel added torch export

chore: use `is_torchdynamo_compiling` instead of `torch._dynamo.is_co…

303f4417

qubvel requested a review from

ArthurZucker 1 year ago

ArthurZucker approved these changes on 2024-12-20

ArthurZucker merged 40292aa4 into main 1 year ago

Reviewers

ArthurZucker

qubvel

Assignees

No one assigned

Labels

torch export

Milestone

No milestone