39
39
type : string
40
40
default : " ['quant', 'tools','restful','pipeline','benchmark','evaluation']"
41
41
schedule :
42
- - cron : ' 00 16 * * 0-4'
42
+ - cron : ' 00 14 * * 0-4'
43
43
44
44
env :
45
45
HOST_PIP_CACHE_DIR : /nvme/github-actions/pip-cache
96
96
runs-on : [self-hosted, linux-a100]
97
97
timeout-minutes : 50
98
98
container :
99
- image : openmmlab/lmdeploy:latest-cu11
99
+ image : openmmlab/lmdeploy:latest-cu12.8
100
100
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
101
101
volumes :
102
102
- /nvme/qa_test_models:/nvme/qa_test_models
@@ -136,7 +136,7 @@ jobs:
136
136
MODELSCOPE_CACHE : /nvme/qa_test_models/modelscope_hub
137
137
MODELSCOPE_MODULES_CACHE : /nvme/qa_test_models/modelscope_modules
138
138
container :
139
- image : openmmlab/lmdeploy:latest-cu11
139
+ image : openmmlab/lmdeploy:latest-cu12.8
140
140
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
141
141
volumes :
142
142
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -152,13 +152,15 @@ jobs:
152
152
cp -r ${{env.TEST_CODE_PATH}}/. .
153
153
- name : Install lmdeploy - dependency
154
154
run : |
155
- python3 -m pip install -r ${{env.OFFLINE_REQUIREMENTS}}
155
+ python3 -m pip install auto_gptq matplotlib attrdict
156
+ python3 -m pip install -r requirements/lite.txt
156
157
- name : Install lmdeploy
157
158
run : |
158
159
python3 -m pip uninstall lmdeploy -y && python3 -m pip install lmdeploy-*.whl --no-deps
159
160
python3 -m pip install -r requirements/test.txt
160
- python3 -m pip install transformers==4.53.1
161
+ rm -rf ${{env.DEEPSEEK_VL}}/build
161
162
pip install ${{env.DEEPSEEK_VL}} --no-deps
163
+ python3 -m pip install transformers==4.53.1 datasets==3.6.0 timm
162
164
- name : Check env
163
165
run : |
164
166
python3 -m pip list
@@ -217,7 +219,7 @@ jobs:
217
219
MODELSCOPE_CACHE : /nvme/qa_test_models/modelscope_hub
218
220
MODELSCOPE_MODULES_CACHE : /nvme/qa_test_models/modelscope_modules
219
221
container :
220
- image : openmmlab/lmdeploy:latest-cu11
222
+ image : openmmlab/lmdeploy:latest-cu12.8
221
223
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
222
224
volumes :
223
225
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -322,7 +324,7 @@ jobs:
322
324
model : Intern-S1
323
325
timeout-minutes : 60
324
326
container :
325
- image : openmmlab/lmdeploy:latest-cu11
327
+ image : openmmlab/lmdeploy:latest-cu12.8
326
328
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
327
329
volumes :
328
330
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -341,6 +343,7 @@ jobs:
341
343
run : |
342
344
python3 -m pip uninstall lmdeploy -y && python3 -m pip install lmdeploy-*.whl --no-deps
343
345
python3 -m pip install -r requirements/test.txt
346
+ rm -rf ${{env.DEEPSEEK_VL}}/build
344
347
pip install ${{env.DEEPSEEK_VL}} --no-deps
345
348
- name : Check env
346
349
run : |
@@ -405,7 +408,7 @@ jobs:
405
408
needs : test_quantization
406
409
timeout-minutes : 120
407
410
container :
408
- image : openmmlab/lmdeploy:latest-cu11
411
+ image : openmmlab/lmdeploy:latest-cu12.8
409
412
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
410
413
volumes :
411
414
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -424,6 +427,7 @@ jobs:
424
427
run : |
425
428
python3 -m pip uninstall lmdeploy -y && python3 -m pip install lmdeploy-*.whl --no-deps
426
429
python3 -m pip install -r requirements/test.txt
430
+ rm -rf ${{env.DEEPSEEK_VL}}/build
427
431
pip install ${{env.DEEPSEEK_VL}} --no-deps
428
432
- name : Check env
429
433
run : |
@@ -461,7 +465,7 @@ jobs:
461
465
needs : test_quantization
462
466
timeout-minutes : 120
463
467
container :
464
- image : openmmlab/lmdeploy:latest-cu11
468
+ image : openmmlab/lmdeploy:latest-cu12.8
465
469
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
466
470
volumes :
467
471
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -480,6 +484,7 @@ jobs:
480
484
run : |
481
485
python3 -m pip uninstall lmdeploy -y && python3 -m pip install lmdeploy-*.whl --no-deps
482
486
python3 -m pip install -r requirements/test.txt
487
+ rm -rf ${{env.DEEPSEEK_VL}}/build
483
488
pip install ${{env.DEEPSEEK_VL}} --no-deps
484
489
- name : Check env
485
490
run : |
@@ -515,7 +520,7 @@ jobs:
515
520
matrix :
516
521
evaluate_type : ['chat', 'base']
517
522
container :
518
- image : openmmlab/lmdeploy:latest-cu11
523
+ image : openmmlab/lmdeploy:latest-cu12.8
519
524
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
520
525
volumes :
521
526
- /nvme/github-actions/pip-cache:/root/.cache/pip
@@ -539,6 +544,7 @@ jobs:
539
544
run : |
540
545
python3 -m pip uninstall lmdeploy -y && python3 -m pip install lmdeploy-*.whl --no-deps
541
546
python3 -m pip install -r requirements/test.txt
547
+ rm -rf ${{env.DEEPSEEK_VL}}/build
542
548
pip install ${{env.DEEPSEEK_VL}} --no-deps
543
549
- name : Install opencompass
544
550
run : |
@@ -588,7 +594,7 @@ jobs:
588
594
timeout-minutes : 5
589
595
runs-on : [self-hosted, linux-a100]
590
596
container :
591
- image : openmmlab/lmdeploy:latest-cu11
597
+ image : openmmlab/lmdeploy:latest-cu12.8
592
598
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
593
599
volumes :
594
600
- /nvme/qa_test_models:/nvme/qa_test_models
@@ -613,7 +619,7 @@ jobs:
613
619
needs : [test_tools, test_restful, test_pipeline, test_benchmark]
614
620
timeout-minutes : 5
615
621
container :
616
- image : openmmlab/lmdeploy:latest-cu11
622
+ image : openmmlab/lmdeploy:latest-cu12.8
617
623
options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e NVIDIA_DISABLE_REQUIRE=1 --pull never"
618
624
volumes :
619
625
- /nvme/github-actions/pip-cache:/root/.cache/pip
0 commit comments