Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d0f11304ab | ||
|
|
b799ebbbee | ||
|
|
24928f5678 |
16
main.py
16
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(),
|
||||||
}
|
}
|
||||||
@@ -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
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user