Commit fc7725b0 authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 686932ce
...@@ -12,7 +12,7 @@ benchmark_wan2.1: ...@@ -12,7 +12,7 @@ benchmark_wan2.1:
- demos - demos
script: script:
- echo "=========================================" - echo "========================================="
- echo "Step 1:Setting up the environment from /workspace/packages/wan2.1" - echo "Step 1 Setting up the environment from /workspace/packages/wan2.1"
- echo "=========================================" - echo "========================================="
# 定义预置文件的根目录 # 定义预置文件的根目录
- export PACKAGE_DIR="/workspace/packages/wan2.1" - export PACKAGE_DIR="/workspace/packages/wan2.1"
...@@ -21,11 +21,10 @@ benchmark_wan2.1: ...@@ -21,11 +21,10 @@ benchmark_wan2.1:
# 设置环境变量,模型路径指向预置目录 # 设置环境变量,模型路径指向预置目录
- 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"
- pip install -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple - pip install -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
- pip install diffusers --no-deps
- ln -s /workspace/packages/wan2.1/Wan_2.1/models/ ./ - ln -s /workspace/packages/wan2.1/Wan_2.1/models/ ./
- echo "=========================================" - echo "========================================="
- echo "Step 2:Starting ComfyUI server" - echo "Step 2 Starting ComfyUI server"
- echo "=========================================" - echo "========================================="
# 创建日志目录 # 创建日志目录
- mkdir -p logs - mkdir -p logs
...@@ -33,17 +32,17 @@ benchmark_wan2.1: ...@@ -33,17 +32,17 @@ benchmark_wan2.1:
- bash run-main.sh > logs/comfyui.log 2>&1 & - bash run-main.sh > logs/comfyui.log 2>&1 &
# 获取ComfyUI进程ID # 获取ComfyUI进程ID
- export COMFYUI_PID=$! - export COMFYUI_PID=$!
- echo "ComfyUI server started with PID: $COMFYUI_PID" - echo "ComfyUI server started with PID $COMFYUI_PID"
- echo "=========================================" - echo "========================================="
- echo "Step 3:Waiting for ComfyUI server to be ready" - echo "Step 3 Waiting for ComfyUI server to be ready"
- echo "=========================================" - echo "========================================="
# 等待服务器启动,最多等待5分钟 # 等待服务器启动,最多等待5分钟
- | - |
timeout=300 timeout=300
count=0 count=0
while [ $count -lt $timeout ]; do while [ $count -lt $timeout ]; do
if grep -q "To see the GUI go to: http://127.0.0.1:8188" logs/comfyui.log; then if grep -q "http://127.0.0.1:8188" logs/comfyui.log; then
echo "ComfyUI server is ready!" echo "ComfyUI server is ready!"
break break
fi fi
...@@ -57,7 +56,7 @@ benchmark_wan2.1: ...@@ -57,7 +56,7 @@ benchmark_wan2.1:
fi fi
- echo "=========================================" - echo "========================================="
- echo "Step 4:Running benchmark tests with different seeds" - echo "Step 4 Running benchmark tests with different seeds"
- echo "=========================================" - echo "========================================="
# 创建结果目录 # 创建结果目录
- mkdir -p results - mkdir -p results
...@@ -79,7 +78,7 @@ benchmark_wan2.1: ...@@ -79,7 +78,7 @@ benchmark_wan2.1:
# 记录结束时间和计算执行时间 # 记录结束时间和计算执行时间
end_time=$(date +%s) end_time=$(date +%s)
duration=$((end_time - start_time)) duration=$((end_time - start_time))
echo "Test 1 execution time: $duration seconds" >> results/performance.log echo "Test 1 execution time $duration seconds" >> results/performance.log
# 等待一段时间确保第一次测试完成 # 等待一段时间确保第一次测试完成
- sleep 10 - sleep 10
...@@ -99,10 +98,10 @@ benchmark_wan2.1: ...@@ -99,10 +98,10 @@ benchmark_wan2.1:
# 记录结束时间和计算执行时间 # 记录结束时间和计算执行时间
end_time=$(date +%s) end_time=$(date +%s)
duration=$((end_time - start_time)) duration=$((end_time - start_time))
echo "Test 2 execution time: $duration seconds" >> results/performance.log echo "Test 2 execution time $duration seconds" >> results/performance.log
- echo "=========================================" - echo "========================================="
- echo "Step 5:Collecting results" - echo "Step 5 Collecting results"
- echo "=========================================" - echo "========================================="
# 查找并复制输出文件 # 查找并复制输出文件
- | - |
...@@ -124,7 +123,7 @@ benchmark_wan2.1: ...@@ -124,7 +123,7 @@ benchmark_wan2.1:
grep "Prompt executed in" results/test2_output.log >> results/summary.txt grep "Prompt executed in" results/test2_output.log >> results/summary.txt
- echo "=========================================" - echo "========================================="
- echo "Step 6:Cleaning up" - echo "Step 6 Cleaning up"
- echo "=========================================" - echo "========================================="
# 停止ComfyUI服务器 # 停止ComfyUI服务器
- kill $COMFYUI_PID || true - kill $COMFYUI_PID || true
......
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