Skip to content

Commit

Permalink
just test
Browse files Browse the repository at this point in the history
  • Loading branch information
wangzhaode committed Sep 22, 2023
1 parent 582ea1a commit a534cd8
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 10 deletions.
8 changes: 4 additions & 4 deletions .github/workflows/build-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ jobs:
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-latest, macos-latest, windows-latest]
os: [windows-latest]

steps:
- uses: actions/checkout@v3
Expand Down Expand Up @@ -55,8 +55,8 @@ jobs:
PACAGE_FILE: ${{ matrix.os }}-package.zip
strategy:
matrix:
os: [ubuntu-latest, macos-latest, windows-latest]
model: [chatglm-6b, chatglm2-6b, codegeex2-6b, qwen-7b-chat, baichuan2-7b-chat, llama2-7b-chat]
os: [windows-latest]
model: [chatglm-6b]

steps:
- uses: actions/download-artifact@v3
Expand All @@ -77,7 +77,7 @@ jobs:
7z x windows-package.zip
cd windows-package
echo ${{ matrix.model }}
./script/model_download.ps1 ${{ matrix.model }}
./script/download_model.ps1 ${{ matrix.model }}
- name: windows-test
if: matrix.os == 'windows-latest'
run: |
Expand Down
15 changes: 9 additions & 6 deletions script/download_model.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,13 @@ param(
)
mkdir $model
cd $model
wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/embedding.mnn -OutFile embedding.mnn
wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/lm.mnn -OutFile lm.mnn
for($i=1; $i -lt 32; $i=$i+1)
{
wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/block_$i.mnn -OutFile block_$i.mnn
$block_num = 28
if ($model.Contains('7b')) {
$block_num = 32
}
cd ..
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) {
Invoke-WebRequest -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/block_$i.mnn -OutFile block_$i.mnn
}
cd ..

0 comments on commit a534cd8

Please sign in to comment.