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,
'command': [
'/bin/bash', '-ic',
'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'
'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'
]
}
},
@@ -321,7 +319,7 @@ def build_config_params(gpu: str) -> str:
},
'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: