Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -67,7 +67,7 @@ benchmark_wan2.1:
       fi
     
     - echo "========================================="
-    - echo "Step 5 Running benchmark tests with file verification"
+    - echo "Step 5 Running benchmark tests with robust verification"
     - echo "========================================="
     # 创建结果目录
     - mkdir -p results
@@ -77,28 +77,44 @@ benchmark_wan2.1:
     # 运行第一次测试（seed=2675441231）
     - echo "Running first test with seed=2675441231"
     - |
+      # 检查原始文件
+      echo "=== Checking original file ==="
+      echo "Original seed-related lines:"
+      grep -n "seed" wan_t2v_14B_1_gpu.py || echo "No seed lines found in original"
+      
       # 创建第一个测试文件
       cp wan_t2v_14B_1_gpu.py temp_test1.py
       
-      # 修改seed值
+      # 尝试多种方法修改seed值
+      echo "=== Attempting seed replacement for test 1 ==="
+      
+      # 方法1: 直接替换args.seed
       sed -i 's/args\.seed/2675441231/g' temp_test1.py
       
-      # 验证修改是否成功
-      echo "=== Verifying temp_test1.py seed modification ==="
+      # 方法2: 如果方法1失败，尝试替换具体的行
+      if ! grep -q "2675441231" temp_test1.py; then
+        echo "Method 1 failed, trying method 2..."
+        sed -i 's/prompt\["3"\]\["inputs"\]\["seed"\] = .*/prompt["3"]["inputs"]["seed"] = 2675441231/g' temp_test1.py
+      fi
+      
+      # 方法3: 如果还是失败，使用更通用的方法
+      if ! grep -q "2675441231" temp_test1.py; then
+        echo "Method 2 failed, trying method 3..."
+        sed -i '/"seed":/s/: .*/: 2675441231,/' temp_test1.py
+      fi
+      
+      # 最终验证
+      echo "=== Final verification for temp_test1.py ==="
+      if grep -q "2675441231" temp_test1.py; then
+        echo "✓ Seed 2675441231 successfully set"
         grep -n "2675441231" temp_test1.py
-      if [ $? -eq 0 ]; then
-        echo "✓ Seed 2675441231 successfully set in temp_test1.py"
       else
-        echo "✗ Failed to set seed 2675441231 in temp_test1.py"
-        echo "Showing relevant lines:"
+        echo "✗ All methods failed to set seed 2675441231"
+        echo "Current seed lines:"
         grep -n "seed" temp_test1.py
         exit 1
       fi
       
-      # 显示修改后的关键行
-      echo "Modified line content:"
-      grep -A2 -B2 "2675441231" temp_test1.py
-      
       # 记录开始时间
       start_time=$(date +%s)
       
@@ -138,25 +154,38 @@ benchmark_wan2.1:
       # 创建第二个测试文件
       cp wan_t2v_14B_1_gpu.py temp_test2.py
       
-      # 修改seed值
+      # 尝试多种方法修改seed值
+      echo "=== Attempting seed replacement for test 2 ==="
+      
+      # 方法1: 直接替换args.seed
       sed -i 's/args\.seed/1234567890/g' temp_test2.py
       
-      # 验证修改是否成功
-      echo "=== Verifying temp_test2.py seed modification ==="
+      # 方法2: 如果方法1失败，尝试替换具体的行
+      if ! grep -q "1234567890" temp_test2.py; then
+        echo "Method 1 failed, trying method 2..."
+        sed -i 's/prompt\["3"\]\["inputs"\]\["seed"\] = .*/prompt["3"]["inputs"]["seed"] = 1234567890/g' temp_test2.py
+      fi
+      
+      # 方法3: 如果还是失败，使用更通用的方法
+      if ! grep -q "1234567890" temp_test2.py; then
+        echo "Method 2 failed, trying method 3..."
+        sed -i '/"seed":/s/: .*/: 1234567890,/' temp_test2.py
+      fi
+      
+      # 最终验证
+      echo "=== Final verification for temp_test2.py ==="
+      if grep -q "1234567890" temp_test2.py; then
+        echo "✓ Seed 1234567890 successfully set"
         grep -n "1234567890" temp_test2.py
-      if [ $? -eq 0 ]; then
-        echo "✓ Seed 1234567890 successfully set in temp_test2.py"
       else
-        echo "✗ Failed to set seed 1234567890 in temp_test2.py"
-        echo "Showing relevant lines:"
+        echo "✗ All methods failed to set seed 1234567890"
+        echo "Current seed lines:"
         grep -n "seed" temp_test2.py
+        echo "Full file content for debugging:"
+        cat temp_test2.py
         exit 1
       fi
       
-      # 显示修改后的关键行
-      echo "Modified line content:"
-      grep -A2 -B2 "1234567890" temp_test2.py
-      
       # 记录开始时间
       start_time=$(date +%s)