dont fallback to cpu for mps backend - SemanticDiff

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

3outeille

Parents

Loading