Skip to content
This repository was archived by the owner on Nov 1, 2024. It is now read-only.

Broadcast json index file rather than loading on all ranks#752

Open
sriniiyer wants to merge 1 commit into
cm3v2_7b_freezefrom
loading_cache_fix
Open

Broadcast json index file rather than loading on all ranks#752
sriniiyer wants to merge 1 commit into
cm3v2_7b_freezefrom
loading_cache_fix

Conversation

@sriniiyer

Copy link
Copy Markdown
Contributor

Patch Description
Broadcast jsonl index instead of loading on all ranks, to prevent slowdown by multiple ranks reading the same file

Testing steps
Tested on 64 nodes, loaded in 2 mins.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants