diff --git a/main.py b/main.py index ec1f15a..67c7af5 100644 --- 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}")