Commit fc7725b0 authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 686932ce
......@@ -12,7 +12,7 @@ benchmark_wan2.1:
- demos
script:
- 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 "========================================="
# 定义预置文件的根目录
- export PACKAGE_DIR="/workspace/packages/wan2.1"
......@@ -21,11 +21,10 @@ benchmark_wan2.1:
# 设置环境变量,模型路径指向预置目录
- 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 diffusers --no-deps
- ln -s /workspace/packages/wan2.1/Wan_2.1/models/ ./
- echo "========================================="
- echo "Step 2:Starting ComfyUI server"
- echo "Step 2 Starting ComfyUI server"
- echo "========================================="
# 创建日志目录
- mkdir -p logs
......@@ -33,17 +32,17 @@ benchmark_wan2.1:
- bash run-main.sh > logs/comfyui.log 2>&1 &
# 获取ComfyUI进程ID
- export COMFYUI_PID=$!
- echo "ComfyUI server started with PID: $COMFYUI_PID"
- echo "ComfyUI server started with PID $COMFYUI_PID"
- echo "========================================="
- echo "Step 3:Waiting for ComfyUI server to be ready"
- echo "Step 3 Waiting for ComfyUI server to be ready"
- echo "========================================="
# 等待服务器启动,最多等待5分钟
- |
timeout=300
count=0
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!"
break
fi
......@@ -57,7 +56,7 @@ benchmark_wan2.1:
fi
- echo "========================================="
- echo "Step 4:Running benchmark tests with different seeds"
- echo "Step 4 Running benchmark tests with different seeds"
- echo "========================================="
# 创建结果目录
- mkdir -p results
......@@ -79,7 +78,7 @@ benchmark_wan2.1:
# 记录结束时间和计算执行时间
end_time=$(date +%s)
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
......@@ -99,10 +98,10 @@ benchmark_wan2.1:
# 记录结束时间和计算执行时间
end_time=$(date +%s)
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 "Step 5:Collecting results"
- echo "Step 5 Collecting results"
- echo "========================================="
# 查找并复制输出文件
- |
......@@ -124,7 +123,7 @@ benchmark_wan2.1:
grep "Prompt executed in" results/test2_output.log >> results/summary.txt
- echo "========================================="
- echo "Step 6:Cleaning up"
- echo "Step 6 Cleaning up"
- echo "========================================="
# 停止ComfyUI服务器
- 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