transformers
Fix: Disable torch.autocast in RotaryEmbedding of Gemma and LLaMa for MPS device
#29439

Merged

Fix: Disable torch.autocast in RotaryEmbedding of Gemma and LLaMa for MPS device #29439

ArthurZucker merged 3 commits into huggingface:main from currybab:fix-gemma-generate

Fix: Disable torch.autocast in RotaryEmbedding of Gemma and LLaMa for…

5a6a8384

ArthurZucker commented on 2024-03-06

Update src/transformers/models/gemma/modeling_gemma.py

3092a650

Update llama ang gemma rope use cpu in mps device

43484d9d

currybab requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2024-03-06

ArthurZucker merged d45f47ab into main 2 years ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone