73782447f2
feat(train): FSDP checkpoint saving ( #3810 )
...
* feat(train): FSDP checkpoint saving
* adding docs for FSDP
* adding a test for the fsdp checkpoint path
* cleanup
* fixing final upload to hub
* refactored initial implementation to use torch fsdp api and adding new tests
2026-06-22 13:51:21 +02:00
df0763a2bc
feat(dependencies): minimal default tag install ( #3362 )
2026-04-12 20:03:04 +02:00
1c388c0002
(Chore) Bump upper bound for torch version ( #2897 )
...
* Bump upper torch version bound
* Apply suggestion from @Copilot
Signed-off-by: Vladislav Sovrasov <vladislav.sovrasov@intel.com >
* Update ref state dicts for schedulers
* Support older than 2.8 torch versions
* Fix precommit
---------
Signed-off-by: Vladislav Sovrasov <vladislav.sovrasov@intel.com >
2026-02-17 23:37:46 +01:00
7cf04a5ec3
chore: move constants to utils ( #2016 )
2025-09-24 11:11:53 +02:00
d4ee470b00
Package folder structure ( #1417 )
...
* Move files
* Replace imports & paths
* Update relative paths
* Update doc symlinks
* Update instructions paths
* Fix imports
* Update grpc files
* Update more instructions
* Downgrade grpc-tools
* Update manifest
* Update more paths
* Update config paths
* Update CI paths
* Update bandit exclusions
* Remove walkthrough section
2025-07-01 16:34:46 +02:00
d8079587a2
Port HIL SERL ( #644 )
...
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co >
Co-authored-by: Eugene Mironov <helper2424@gmail.com >
Co-authored-by: s1lent4gnt <kmeftah.khalil@gmail.com >
Co-authored-by: Ke Wang <superwk1017@gmail.com >
Co-authored-by: Yoel Chornton <yoel.chornton@gmail.com >
Co-authored-by: imstevenpmwork <steven.palma@huggingface.co >
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co >
2025-06-13 13:15:47 +02:00
0cf864870c
[Fix] Unpin torch beyond 2.6.0 & torchcodec beyond 0.2.1 ( #1127 )
2025-05-28 16:54:20 +02:00
974028bd28
Organize test folders ( #856 )
...
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org >
2025-03-13 14:05:55 +01:00
Maxime Ellerbach