transformers
a6adc05e
- symbolic_trace: add past_key_values, llama, sdpa support (#28447)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
symbolic_trace: add past_key_values, llama, sdpa support (#28447) * torch.fx: add pkv, llama, sdpa support * Update src/transformers/models/opt/modeling_opt.py * remove spaces * trigger ci * use explicit variable names
References
#28447 - symbolic_trace: add past_key_values, llama, sdpa support
Author
fxmarty
Parents
09eb11a1
Loading