model: support deepseek-ocr (#11891)
Co-authored-by: yhyang201 <47235274+yhyang201@users.noreply.github.com> Co-authored-by: yhyang201 <yhyang201@gmail.com> Co-authored-by: Shi Shuai <126407087+shuaills@users.noreply.github.com> Co-authored-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
This commit is contained in:
@@ -32,6 +32,7 @@ class TestOpenAIMLLMServerBase(CustomTestCase):
|
||||
model: str
|
||||
extra_args: list = []
|
||||
fixed_args: list = ["--trust-remote-code", "--enable-multimodal"]
|
||||
trust_remote_code: bool = True
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
@@ -42,7 +43,11 @@ class TestOpenAIMLLMServerBase(CustomTestCase):
|
||||
cls.base_url,
|
||||
timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||
api_key=cls.api_key,
|
||||
other_args=cls.extra_args + cls.fixed_args,
|
||||
other_args=(
|
||||
cls.extra_args + cls.fixed_args + ["--trust-remote-code"]
|
||||
if cls.trust_remote_code
|
||||
else []
|
||||
),
|
||||
)
|
||||
cls.base_url += "/v1"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user