transformers
5f1c05cf - [loading][TP] Fix device placement at loading-time, and simplify sharding primitives (#43003)

Commit

6 days ago

[loading][TP] Fix device placement at loading-time, and simplify sharding primitives (#43003) * why was it so complicated * device * simplify * fix * imports * simplify quite a bit * fix

References

#43003 - [loading][TP] Fix device placement at loading-time, and simplify sharding primitives

Author

Cyrilvallez

Parents

e9f0f8e0

transformers 5f1c05cf - [loading][TP] Fix device placement at loading-time, and simplify sharding primitives (#43003)

transformers
5f1c05cf - [loading][TP] Fix device placement at loading-time, and simplify sharding primitives (#43003)