PR #30273 Fix RecurrentGemma device_map

Fix RecurrentGemma device_map #30273

ArthurZucker merged 3 commits into huggingface:main from SunMarc:fix-buffer-recurrent-gemma

Switch to non persistant buffer

7d0595d4

fix device mismatch issue due to cache

b95c44e9

style

3b6d19c2

SunMarc requested a review from

ArthurZucker 1 year ago

SunMarc changed the title ~~Fix recurrent gemma device_map~~ Fix RecurrentGemma device_map 1 year ago

SunMarc commented on 2024-04-16

ArthurZucker commented on 2024-04-16

ArthurZucker approved these changes on 2024-04-18

ArthurZucker merged 7509a0ad into main 1 year ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone