Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -67,7 +67,7 @@ benchmark_wan2.1:
       fi
     
     - echo "========================================="
-    - echo "Step 5 Running benchmark tests with robust verification"
+    - echo "Step 5 Running benchmark tests with precise seed replacement"
     - echo "========================================="
     # 创建结果目录
     - mkdir -p results
@@ -77,39 +77,19 @@ 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值
-      echo "=== Attempting seed replacement for test 1 ==="
-      
-      # 方法1: 直接替换args.seed
-      sed -i 's/args\.seed/2675441231/g' temp_test1.py
-      
-      # 方法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
+      # 精确替换JSON中的seed值
+      sed -i 's/"seed": [0-9]\+,/"seed": 2675441231,/' temp_test1.py
       
-      # 最终验证
-      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
+      # 验证修改是否成功
+      echo "=== Verifying temp_test1.py seed modification ==="
+      if grep -q '"seed": 2675441231,' temp_test1.py; then
+        echo "✓ Seed 2675441231 successfully set in temp_test1.py"
+        grep -n '"seed": 2675441231,' temp_test1.py
       else
-        echo "✗ All methods failed to set seed 2675441231"
+        echo "✗ Failed to set seed 2675441231 in temp_test1.py"
         echo "Current seed lines:"
         grep -n "seed" temp_test1.py
         exit 1
@@ -154,37 +134,44 @@ benchmark_wan2.1:
       # 创建第二个测试文件
       cp wan_t2v_14B_1_gpu.py temp_test2.py
       
-      # 尝试多种方法修改seed值
-      echo "=== Attempting seed replacement for test 2 ==="
+      # 精确替换JSON中的seed值
+      sed -i 's/"seed": [0-9]\+,/"seed": 1234567890,/' temp_test2.py
       
-      # 方法1: 直接替换args.seed
-      sed -i 's/args\.seed/1234567890/g' temp_test2.py
+      # 验证修改是否成功
+      echo "=== Verifying temp_test2.py seed modification ==="
+      if grep -q '"seed": 1234567890,' temp_test2.py; then
+        echo "✓ Seed 1234567890 successfully set in temp_test2.py"
+        grep -n '"seed": 1234567890,' temp_test2.py
+      else
+        echo "✗ Failed to set seed 1234567890 in temp_test2.py"
+        echo "Current seed lines:"
+        grep -n "seed" temp_test2.py
+        echo "Attempting alternative replacement method..."
         
-      # 方法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
+        # 备用方法：使用Python直接修改
+        python3 -c "
+import re
+with open('temp_test2.py', 'r') as f:
+    content = f.read()
     
-      # 方法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
+# 使用正则表达式替换seed值
+content = re.sub(r'(\"seed\":\s*)\d+', r'\g<1>1234567890', content)
+
+with open('temp_test2.py', 'w') as f:
+    f.write(content)
+"
         
-      # 最终验证
-      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 grep -q '"seed": 1234567890,' temp_test2.py; then
+          echo "✓ Seed 1234567890 successfully set using Python method"
+          grep -n '"seed": 1234567890,' temp_test2.py
         else
           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
+      fi
       
       # 记录开始时间
       start_time=$(date +%s)