diff --git a/script/download_model.ps1 b/script/download_model.ps1 index 38e6a027..78f72df0 100644 --- a/script/download_model.ps1 +++ b/script/download_model.ps1 @@ -3,17 +3,17 @@ param( ) mkdir $model cd $model -$block_num = 28 +$block_num = 27 if ($model.Contains('7b')) { - $block_num = 32 + $block_num = 31 } if ($model.Contains('1.8b')) { - $block_num = 24 + $block_num = 23 } Invoke-WebRequest -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/tokenizer.txt -OutFile tokenizer.txt Invoke-WebRequest -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/embedding.mnn -OutFile embedding.mnn Invoke-WebRequest -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/lm.mnn -OutFile lm.mnn -for ($i=1; $i -lt $block_num; $i=$i+1) { +for ($i=0; $i -lt $block_num; $i=$i+1) { Invoke-WebRequest -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/block_$i.mnn -OutFile block_$i.mnn } cd .. diff --git a/script/download_model.sh b/script/download_model.sh index ae720134..82ae78d6 100755 --- a/script/download_model.sh +++ b/script/download_model.sh @@ -13,7 +13,7 @@ if [ $is_7b ]; then block_num=31 fi if [ $is_1_8b ]; then - block_num=24 + block_num=23 fi # download models wget -c -nv https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/tokenizer.txt @@ -23,4 +23,4 @@ for i in `seq 0 $block_num` do wget -c -nv https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/block_$i.mnn done -cd .. \ No newline at end of file +cd ..