Skip to content

Update llama.sh#2

Open
WqyJh wants to merge 2 commits intoshawwn:mainfrom
WqyJh:patch-1
Open

Update llama.sh#2
WqyJh wants to merge 2 commits intoshawwn:mainfrom
WqyJh:patch-1

Conversation

@WqyJh
Copy link

@WqyJh WqyJh commented Mar 6, 2023

Use aria2c to for parallel downloading and continuation

Use aria2c to for parallel downloading and continuation
llama.sh Outdated
for s in $(seq -f "0%g" 0 ${N_SHARD_DICT[$i]})
do
wget ${PRESIGNED_URL/'*'/"${i}/consolidated.${s}.pth"} -O ${TARGET_FOLDER}"/${i}/consolidated.${s}.pth"
aria2c -c -s10 -x10 -k1M ${PRESIGNED_URL/'*'/"${i}/consolidated.${s}.pth"} -o ${TARGET_FOLDER}"/${i}/consolidated.${s}.pth"
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

maybe could add a check with "which aria2c", if it fails, use wget as a fallback.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants