PR #30775 Make `Gemma` work with `torch.compile`

Make `Gemma` work with `torch.compile` #30775

ydshieh merged 12 commits into main from fix_gemma_torch_compile

ydshieh commented on 2024-05-13

ydshieh marked this pull request as ready for review 1 year ago

ydshieh requested a review from

ArthurZucker 1 year ago

ydshieh added run-slow

ArthurZucker commented on 2024-05-13

ArthurZucker commented on 2024-05-14

fix

12059553

[run-slow] gemma

afcd5ca8

ydshieh force pushed 1 year ago

ydshieh marked this pull request as draft 1 year ago

add test

a63c3ba1

ydshieh force pushed to a63c3ba1 1 year ago

ydshieh marked this pull request as ready for review 1 year ago

ydshieh marked this pull request as draft 1 year ago

ydshieh marked this pull request as ready for review 1 year ago

ydshieh requested a review from

ArthurZucker 1 year ago

ArthurZucker commented on 2024-05-15

add `test_compile_static_cache`

b97195de

fix

f3e622e2

style

b60f634c

remove subprocess

fd287405

use attribute

a1dbbadf

fix

9358e16e

style

db054240

ydshieh requested a review from

ArthurZucker 1 year ago

ArthurZucker approved these changes on 2024-05-15

update

3280eea3

[run-slow] dbrx,gemma,jetmoe,phi3,recurrent_gemma

e55d24b7

ydshieh merged 1b3dba94 into main 1 year ago

ydshieh deleted the fix_gemma_torch_compile branch 1 year ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

run-slow

Milestone

No milestone

transformers Make `Gemma` work with `torch.compile` #30775 Merged

Make `Gemma` work with `torch.compile` #30775

transformers
Make `Gemma` work with `torch.compile`
#30775

Merged