ministral3 (#14251)

Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
Co-authored-by: Yueming Yuan <yy28@illinois.edu>
This commit is contained in:
Xinyuan Tong
2025-12-04 14:31:26 -08:00
committed by GitHub
co-authored by Yueming Yuan
parent c1006fd8a1
commit 6d37e70883
14 changed files with 245 additions and 26 deletions
+8 -8
View File
@@ -83,9 +83,9 @@ async def process_sample(
assert image is not None
image_path = sample["image_path"]
extra_body = None if lora_path is None else {"lora_path": lora_path}
response = await client.chat.completions.create(
model="default",
messages=[
payload = {
"model": "default",
"messages": [
{
"role": "user",
"content": [
@@ -95,11 +95,11 @@ async def process_sample(
],
}
],
temperature=0,
max_completion_tokens=sampling_params["max_new_tokens"],
max_tokens=sampling_params["max_new_tokens"],
extra_body=extra_body,
)
"extra_body": extra_body,
}
if sampling_params:
payload.update(sampling_params)
response = await client.chat.completions.create(**payload)
return sample, response.choices[0].message.content