2 Commits

Author SHA1 Message Date
z3st
24928f5678 fix: re-add GGUF to exclusion (vllm cannot load GGUF models) 2026-07-25 13:36:17 +08:00
z3st
3eed33f0d8 chore: increase submit limit from 2 to 5 2026-07-24 20:30:10 +08:00

10
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"""
init_db()
log("=" * 50)
@@ -344,7 +344,7 @@ def run_pipeline(submit_limit: int = 2):
log("\n--- 阶段2: 格式筛选 ---")
hf_models = []
format_skipped = 0
SKIP_FORMATS = ['GPTQ', 'AWQ']
SKIP_FORMATS = ['GGUF', 'GPTQ', 'AWQ']
for m in all_models:
mid_upper = m['model_id'].upper()
skip = False
@@ -356,7 +356,7 @@ def run_pipeline(submit_limit: int = 2):
break
if not skip:
hf_models.append(m)
log(f"格式筛选: {len(hf_models)} 通过, {format_skipped} 跳过 (GPTQ/AWQ)")
log(f"格式筛选: {len(hf_models)} 通过, {format_skipped} 跳过 (GGUF/GPTQ/AWQ)")
# 3. 架构筛选只保留有标准config.json的模型排除无效格式
log("\n--- 阶段3: 架构检查 ---")
@@ -487,7 +487,7 @@ class AgentHandler(BaseHTTPRequestHandler):
if content_len > 0:
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})
@@ -704,7 +704,7 @@ def main():
try:
state['running'] = True
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}
except Exception as e:
log(f"流程异常: {traceback.format_exc()}")