3 Commits

Author SHA1 Message Date
z3st
d0f11304ab feat: switch framework to vllm_fix_tokenizer 2026-07-25 23:04:39 +08:00
z3st
b799ebbbee fix: cancel all tasks across ALL GPUs, not just TARGET_GPU 2026-07-25 22:57:42 +08:00
z3st
24928f5678 fix: re-add GGUF to exclusion (vllm cannot load GGUF models) 2026-07-25 13:36:17 +08:00

16
main.py
View File

@@ -247,10 +247,10 @@ def check_queue_available() -> int:
def build_config_params() -> str: def build_config_params() -> str:
"""构建 YAML 配置 - vllm""" """构建 YAML 配置 - vllm_fix_tokenizer"""
params = { params = {
'framework': 'vllm', 'framework': 'vllm_fix_tokenizer',
'nv_framework': 'vllm', 'nv_framework': 'vllm_fix_tokenizer',
'api': 'completion', 'api': 'completion',
'max_tokens': 1024, 'max_tokens': 1024,
'temperature': 0.7, 'temperature': 0.7,
@@ -295,7 +295,7 @@ def submit_model(model_url: str) -> tuple:
'modelAddress': normalize_model_url(model_url), 'modelAddress': normalize_model_url(model_url),
'taskType': 'text-generation', 'taskType': 'text-generation',
'targetGpu': TARGET_GPU, 'targetGpu': TARGET_GPU,
'framework': 'vllm', 'framework': 'vllm_fix_tokenizer',
'strategyId': STRATEGY_ID, 'strategyId': STRATEGY_ID,
'configParams': build_config_params(), 'configParams': build_config_params(),
} }
@@ -344,7 +344,7 @@ def run_pipeline(submit_limit: int = 5):
log("\n--- 阶段2: 格式筛选 ---") log("\n--- 阶段2: 格式筛选 ---")
hf_models = [] hf_models = []
format_skipped = 0 format_skipped = 0
SKIP_FORMATS = ['GPTQ', 'AWQ'] SKIP_FORMATS = ['GGUF', 'GPTQ', 'AWQ']
for m in all_models: for m in all_models:
mid_upper = m['model_id'].upper() mid_upper = m['model_id'].upper()
skip = False skip = False
@@ -356,7 +356,7 @@ def run_pipeline(submit_limit: int = 5):
break break
if not skip: if not skip:
hf_models.append(m) 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的模型排除无效格式 # 3. 架构筛选只保留有标准config.json的模型排除无效格式
log("\n--- 阶段3: 架构检查 ---") log("\n--- 阶段3: 架构检查 ---")
@@ -568,7 +568,7 @@ class AgentHandler(BaseHTTPRequestHandler):
'modelAddress': 'https://www.modelscope.cn/models/Qwen/Qwen3-8B', 'modelAddress': 'https://www.modelscope.cn/models/Qwen/Qwen3-8B',
'taskType': 'text-generation', 'taskType': 'text-generation',
'targetGpu': TARGET_GPU, 'targetGpu': TARGET_GPU,
'framework': 'vllm', 'framework': 'vllm_fix_tokenizer',
'strategyId': STRATEGY_ID, 'strategyId': STRATEGY_ID,
'configParams': build_config_params(), 'configParams': build_config_params(),
}, },
@@ -686,7 +686,7 @@ def main():
json={ json={
'modelAddress': 'https://www.modelscope.cn/models/Qwen/Qwen3-8B', 'modelAddress': 'https://www.modelscope.cn/models/Qwen/Qwen3-8B',
'taskType': 'text-generation', 'targetGpu': TARGET_GPU, 'taskType': 'text-generation', 'targetGpu': TARGET_GPU,
'framework': 'vllm', 'strategyId': STRATEGY_ID, 'framework': 'vllm_fix_tokenizer', 'strategyId': STRATEGY_ID,
'configParams': build_config_params(), 'configParams': build_config_params(),
}, timeout=10 }, timeout=10
) )