2 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

12
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(),
} }
@@ -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
) )