Commit 9d26048c authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 460977ed
# 定义流水线要使用的 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: harbor.sourcefind.cn:5443/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.4-1230-py3.10-20260115
# 只定义一个阶段 # 只定义一个阶段
stages: stages:
...@@ -39,8 +39,8 @@ benchmark_sdxl: ...@@ -39,8 +39,8 @@ benchmark_sdxl:
- echo "Step 3:Running the benchmark" - echo "Step 3: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"
- export LD_LIBRARY_PATH="$CI_PROJECT_DIR/package-miopen-dev-0801-ubuntu20/lib/:$LD_LIBRARY_PATH" #- export LD_LIBRARY_PATH="$CI_PROJECT_DIR/package-miopen-dev-0801-ubuntu20/lib/:$LD_LIBRARY_PATH"
- export PYTORCH_MIOPEN_SUGGEST_NHWC=1 - export PYTORCH_MIOPEN_SUGGEST_NHWC=1
- export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH" - export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH"
- export MODEL_PATH="$PACKAGE_DIR/stable-diffusion-xl-base-1.0" - export MODEL_PATH="$PACKAGE_DIR/stable-diffusion-xl-base-1.0"
......
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