PR #31247 Add torch.compile Support For Mamba

Add torch.compile Support For Mamba #31247

zhenglongjiepheonix merged 41 commits into huggingface:main from zhenglongjiepheonix:longjie/add_torch_compile_for_mamba

modify mamba cache

3f9eeb8e

ArthurZucker commented on 2024-06-05

set up cache

d6413fef

add test

054d7cfc

fix conflict

93fa82fc

[run-slow] mamba

48e0ff06

zhenglongjiepheonix commented on 2024-06-06

zhenglongjiepheonix requested a review from

ArthurZucker 2 years ago

zhenglongjiepheonix added run-slow

[run-slow] mamba

1e6f0e90

zhenglongjiepheonix changed the title ~~[WIP] Add torch.compile Support For Mamba~~ Add torch.compile Support For Mamba 1 year ago

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

23d383a3

ArthurZucker commented on 2024-06-07

address comments

49ee4cbc

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

bc85aa99

[run-slow] mamba

247e7894

zhenglongjiepheonix requested a review from

ArthurZucker 1 year ago

ArthurZucker commented on 2024-06-10

use_cache_position

9e1fb0e0

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

8a132ac8

[run-slow] mamba

100b9991

[run-slow] mamba

2dc89867

zhenglongjiepheonix requested a review from

ArthurZucker 1 year ago

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

9732a62a

[run-slow] mamba

c2c8e5d2

ArthurZucker commented on 2024-06-12

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

9120fbfc

[run-slow] mamba

63568dfd

zhenglongjiepheonix requested a review from

ArthurZucker 1 year ago

fix

0736da8b

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

3141d26f

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

584528ef

cache in generate

f136d3c8

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

ec4a7a36

[run-slow] mamba

400feec6

zhenglongjiepheonix commented on 2024-06-20

ArthurZucker commented on 2024-06-26

address comments

de9182d4

resolve conflict

38441cd7

[run-slow] mamba

a254a098

zhenglongjiepheonix requested a review from

ArthurZucker 1 year ago

[run-slow] mamba

ac456edf

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

3e958136

address comments

801e8d15

[run-slow] mamba

b8af2a32

fix

b2e8a0b4

fix

708d3022

[run-slow] mamba

97b1add3

fix

2feeeb09

ArthurZucker approved these changes on 2024-07-09

fix conflict

b045c38b

[run-slow] mamba

fcdc98f4

gante approved these changes on 2024-07-16

fix cache name

82b0c9b0

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

611e0d7d

Merge remote-tracking branch 'upstream/main' into longjie/add_torch_c…

bc1563ed

[run-slow] mamba

78c8c1ca

zhenglongjiepheonix merged c75969ee into main 1 year ago

Reviewers

ArthurZucker

gante

Assignees

No one assigned

Labels

run-slow

Milestone

No milestone

transformers Add torch.compile Support For Mamba #31247 Merged

Add torch.compile Support For Mamba #31247

transformers
Add torch.compile Support For Mamba
#31247

Merged