fix: replace gpu variable with TARGET_GPU in submit record

2026-07-24 00:02:35 +08:00
1 changed files with 1 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -385,7 +385,7 @@ def run_pipeline(submit_limit: int = 30):
            log(f"  ✅ {m['model_id']}")
            db_conn.execute(
                'INSERT OR REPLACE INTO submitted VALUES (?,?,?,?)',
-                (m['model_id'], gpu, str(task_id), datetime.now().isoformat())
+                (m['model_id'], TARGET_GPU, str(task_id), datetime.now().isoformat())
            )
        else:
            log(f"  ❌ {m['model_id']}: {msg}")