Commit 5891a84c authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 0a7b3572
# 定义流水线要使用的 Docker 镜像 # 定义流水线要使用的 Docker 镜像
image: image.sourcefind.cn:5000/dcu/admin/base/vllm:0.8.5-ubuntu22.04-dtk25.04.1-rc5-das1.6-py3.10-20250724 image: image.sourcefind.cn:5000/dcu/admin/base/vllm:0.8.5-ubuntu22.04-dtk25.04.1-rc5-das1.6-py3.10-20250724
# 只定义一个阶段 # 只定义一个阶段
stages: stages:
- test - test
...@@ -13,7 +12,7 @@ benchmark_sdxl: ...@@ -13,7 +12,7 @@ benchmark_sdxl:
- demos - demos
script: script:
- echo "=========================================" - echo "========================================="
- echo "Step 1: Checking and downloading assets" - echo "Step 1:Checking and downloading assets"
- echo "=========================================" - echo "========================================="
# 条件性下载模型 # 条件性下载模型
- | - |
...@@ -40,7 +39,7 @@ benchmark_sdxl: ...@@ -40,7 +39,7 @@ benchmark_sdxl:
fi fi
- echo "=========================================" - echo "========================================="
- echo "Step 2: Setting up the environment" - echo "Step 2:Setting up the environment"
- echo "=========================================" - echo "========================================="
# 解压库文件 # 解压库文件
- tar -xzf "rocblas-install-0626-bug.tar.gz" - tar -xzf "rocblas-install-0626-bug.tar.gz"
...@@ -56,13 +55,13 @@ benchmark_sdxl: ...@@ -56,13 +55,13 @@ benchmark_sdxl:
- cd .. - cd ..
- echo "=========================================" - echo "========================================="
- echo "Step 3: Installing Python packages" - echo "Step 3:Installing Python packages"
- echo "=========================================" - echo "========================================="
# 安装 Python 包 (有缓存时会非常快) # 安装 Python 包 (有缓存时会非常快)
- pip install "diffusers-0.33.1-py3-none-any.whl" "lightop-0.5.0+das.dtk25041-cp310-cp310-linux_x86_64.whl" "litserve-0.2.15-py3-none-any.whl" -q - pip install "diffusers-0.33.1-py3-none-any.whl" "lightop-0.5.0+das.dtk25041-cp310-cp310-linux_x86_64.whl" "litserve-0.2.15-py3-none-any.whl" -q
- echo "=========================================" - echo "========================================="
- echo "Step 4: Running the benchmark" - echo "Step 4:Running the benchmark"
- echo "=========================================" - echo "========================================="
# 设置环境变量并运行测试 # 设置环境变量并运行测试
- export LD_LIBRARY_PATH="$CI_PROJECT_DIR/rocblas-install/lib/:$LD_LIBRARY_PATH" - export LD_LIBRARY_PATH="$CI_PROJECT_DIR/rocblas-install/lib/:$LD_LIBRARY_PATH"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment