fix: replace gpu variable with TARGET_GPU in submit record

This commit is contained in:
z3st
2026-07-24 00:02:35 +08:00
parent 8e8fd50927
commit 68c7a99e68

View File

@@ -385,7 +385,7 @@ def run_pipeline(submit_limit: int = 30):
log(f"{m['model_id']}") log(f"{m['model_id']}")
db_conn.execute( db_conn.execute(
'INSERT OR REPLACE INTO submitted VALUES (?,?,?,?)', '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: else:
log(f"{m['model_id']}: {msg}") log(f"{m['model_id']}: {msg}")