transformers
symbolic_trace: add past_key_values, llama, sdpa support
#28447
Merged
Overview
Commits
6
Changes
View On GitHub

symbolic_trace: add past_key_values, llama, sdpa support #28447

fxmarty merged 6 commits into huggingface:main from fxmarty:fx-symbolic-trace-pkv-support
fxmarty
fxmarty torch.fx: add pkv, llama, sdpa support
6f17cf42
fxmarty
fxmarty commented on 2024-01-11
fxmarty Update src/transformers/models/opt/modeling_opt.py
12d6d100
fxmarty remove spaces
f17350a0
fxmarty fxmarty requested a review from michaelbenayoun michaelbenayoun 2 years ago
michaelbenayoun
michaelbenayoun approved these changes on 2024-01-12
fxmarty Merge branch 'main' into fx-symbolic-trace-pkv-support
41fb1984
fxmarty fxmarty requested a review from amyeroberts amyeroberts 1 year ago
fxmarty fxmarty requested a review from ArthurZucker ArthurZucker 1 year ago
HuggingFaceDocBuilderDev
fxmarty trigger ci
9d1b317c
amyeroberts
amyeroberts approved these changes on 2024-01-16
fxmarty use explicit variable names
c58f4a5e
fxmarty fxmarty merged a6adc05e into main 1 year 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