transformers
make LlamaModel._update_causal_mask torch compilable
#35187
Merged
Overview
Commits
3
Changes
View On GitHub

make LlamaModel._update_causal_mask torch compilable #35187

ArthurZucker merged 3 commits into huggingface:main from winglian:torch-compile-llama-attn
winglian
Rocketknight1
winglian
Rocketknight1
ArthurZucker
ArthurZucker approved these changes on 2024-12-23
winglian make LlamaModel._update_causal_mask torch compilable
58bb76a7
winglian chore: lint (make fix-copies)
b40f041a
ArthurZucker ArthurZucker force pushed to b40f041a 362 days ago
ArthurZucker fix-copies
e0a731d7
ArthurZucker
ArthurZucker ArthurZucker merged 5e7aedeb into main 362 days ago
winglian winglian deleted the torch-compile-llama-attn branch 355 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum