transformers
c5415974
- dont fallback to cpu for mps backend
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
11 days ago
dont fallback to cpu for mps backend
References
#44083 - FSDP2 native support in transformers
#44974 - Refactor core_model_loading to support FSDP shard-on-read loading
#45028 - TP refactor for FSDP + TP integration
Author
3outeille
Parents
a6593ace
Loading