z3st/modelhub-submit-agent
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: yaml.dump width=1000 to prevent command string wrapping

Browse Source
This commit is contained in:
z3st
2026-07-22 22:43:35 +08:00
parent b3b52848c2
commit 85c456afe0
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -302,9 +302,7 @@ def build_config_params(gpu: str) -> str:
'gpu_num': 1, 'gpu_num': 1,
'command': [ 'command': [
'/bin/bash', '-ic', '/bin/bash', '-ic',
'vllm serve /model --port 8000 --served-model-name llm ' 'vllm serve /model --port 8000 --served-model-name llm --max-model-len 2048 --dtype auto --gpu-memory-utilization 0.95 -tp 1 --enforce-eager --trust-remote-code'
'--max-model-len 2048 --dtype auto --gpu-memory-utilization 0.95 '
'-tp 1 --enforce-eager --trust-remote-code'
] ]
} }
}, },
@@ -321,7 +319,7 @@ def build_config_params(gpu: str) -> str:
}, },
'model': 'llm', 'model': 'llm',
} }
return yaml.dump(params, default_flow_style=False, allow_unicode=True) return yaml.dump(params, default_flow_style=False, allow_unicode=True, width=1000)
def submit_model(model_url: str, gpu: str, token: str) -> tuple: def submit_model(model_url: str, gpu: str, token: str) -> tuple: