Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d0f11304ab | ||
|
|
b799ebbbee |
12
main.py
12
main.py
@@ -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
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user