Matej Sirovatka 46a4b7c909 Feat: save_pretrained for tensor parallel (and other parallelisms) models (#37919) ...

* tmp: initial save pretrained with dtensors

* Feat: add correctness tests

* Refactor: version checks

* Temp: 1:1 checkpoint llama4

* refactor

* Tests

* Feat: works

* Style

* Feat: version checks + minor fixes

* Style

* Fix: version checks in tests

* Feat: move more stuff into tensor_parallel.py

2025-05-19 18:16:21 +00:00

..

test_tensor_parallel.py

Feat: save_pretrained for tensor parallel (and other parallelisms) models (#37919)

2025-05-19 18:16:21 +00:00