transformers
Llama 3.1: replace for loop by tensor ops at inv_freq initialization
#32244

Merged

Llama 3.1: replace for loop by tensor ops at inv_freq initialization #32244

gante merged 2 commits into huggingface:main from gante:llama_31_rope_tensor_ops

replace for loop by tensor ops

c5c3f3d1

gante requested a review from

ArthurZucker 2 years ago

gante changed the title ~~Llama 3.1: replace for loop by tensor ops at inv_freq initializatoin~~ Llama 3.1: replace for loop by tensor ops at inv_freq initialization 2 years ago

ArthurZucker approved these changes on 2024-07-26

rm assert; readability

e2b79a83

gante merged 44f6fdd7 into main 2 years ago

gante deleted the llama_31_rope_tensor_ops branch 2 years ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Llama 3.1: replace for loop by tensor ops at inv_freq initialization #32244 Merged

Llama 3.1: replace for loop by tensor ops at inv_freq initialization #32244

transformers
Llama 3.1: replace for loop by tensor ops at inv_freq initialization
#32244

Merged