Merge pull request #177 from aleksandar-babic/main

fix: added back limit_mm_per_prompt to engine args
This commit is contained in:
Marut Pandya
2025-05-03 21:01:58 -07:00
committed by GitHub
2 changed files with 10 additions and 1 deletions
+1
View File
@@ -91,6 +91,7 @@ DEFAULT_ARGS = {
"disable_logprobs_during_spec_decoding": os.getenv('DISABLE_LOGPROBS_DURING_SPEC_DECODING', None),
"otlp_traces_endpoint": os.getenv('OTLP_TRACES_ENDPOINT', None),
"use_v2_block_manager": os.getenv('USE_V2_BLOCK_MANAGER', 'true'),
"limit_mm_per_prompt": convert_limit_mm_per_prompt(os.getenv('LIMIT_MM_PER_PROMPT', "image=1")),
}
def match_vllm_args(args):
+9 -1
View File
@@ -86,7 +86,7 @@
"TYPICAL_ACCEPTANCE_SAMPLER_POSTERIOR_THRESHOLD", "TYPICAL_ACCEPTANCE_SAMPLER_POSTERIOR_ALPHA",
"MODEL_LOADER_EXTRA_CONFIG", "PREEMPTION_MODE", "PREEMPTION_CHECK_PERIOD",
"PREEMPTION_CPU_CAPACITY", "MAX_LOG_LEN", "DISABLE_LOGGING_REQUEST",
"ENABLE_AUTO_TOOL_CHOICE", "TOOL_CALL_PARSER"
"ENABLE_AUTO_TOOL_CHOICE", "TOOL_CALL_PARSER", "LIMIT_MM_PER_PROMPT"
]
},
{
@@ -1324,6 +1324,14 @@
{ "value": "pythonic", "label": "Pythonic" },
{ "value": "internlm", "label": "InternLM" }
]
},
"LIMIT_MM_PER_PROMPT": {
"env_var_name": "LIMIT_MM_PER_PROMPT",
"value": "image=1",
"title": "Limit MM Per Prompt",
"description": "Limit MM per prompt",
"required": false,
"type": "text"
}
}
}