z3st/modelhub-submit-agent
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

chore: increase submit limit from 2 to 5

Browse Source
This commit is contained in:
z3st
2026-07-24 20:30:10 +08:00
parent a2ef82d954
commit 3eed33f0d8
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -314,7 +314,7 @@ def submit_model(model_url: str) -> tuple:
# 主流程 # 主流程
# ============================================================ # ============================================================
def run_pipeline(submit_limit: int = 2): def run_pipeline(submit_limit: int = 5):
"""完整流程：搜索→筛选→提交（只针对目标GPU）""" """完整流程：搜索→筛选→提交（只针对目标GPU）"""
init_db() init_db()
log("=" * 50) log("=" * 50)
@@ -487,7 +487,7 @@ class AgentHandler(BaseHTTPRequestHandler):
if content_len > 0: if content_len > 0:
body = json.loads(self.rfile.read(content_len)) body = json.loads(self.rfile.read(content_len))
limit = body.get('limit', 2) limit = body.get('limit', 5)
self._json({'status': 'started', 'gpu': TARGET_GPU, 'limit': limit}) self._json({'status': 'started', 'gpu': TARGET_GPU, 'limit': limit})
@@ -704,7 +704,7 @@ def main():
try: try:
state['running'] = True state['running'] = True
state['last_run'] = datetime.now().isoformat() state['last_run'] = datetime.now().isoformat()
count = run_pipeline(submit_limit=2) count = run_pipeline(submit_limit=5)
state['last_result'] = {'submitted': count, 'success': True} state['last_result'] = {'submitted': count, 'success': True}
except Exception as e: except Exception as e:
log(f"流程异常: {traceback.format_exc()}") log(f"流程异常: {traceback.format_exc()}")