fix: load HF model only on rank 0#544

Merged

parthchadha merged 19 commits intomainfrom

pchadha/large-model-state-dict-load

Jul 2, 2025

Commits on Jun 18, 2025

Use set_model_state_dict and load model on rank 0
parthchadha
committed

Commits on Jun 24, 2025

Merge remote-tracking branch 'origin/main' into pchadha/large-model-state-dict-load
parthchadha
committed
Fix use of model_config and remove duplicate args
parthchadha
committed
Merge remote-tracking branch 'origin/main' into pchadha/large-model-state-dict-load
parthchadha
committed

Commits on Jun 26, 2025

Commits on Jun 27, 2025

Commits on Jul 1, 2025