Spaces:
Paused
Paused
Peter Larnholt
commited on
Commit
·
55381a2
1
Parent(s):
1b713bd
Enable auto tool choice for better tool calling support
Browse files
app.py
CHANGED
|
@@ -26,6 +26,7 @@ VLLM_ARGS = [
|
|
| 26 |
"--max-model-len", "8192", # fits A10G 24GB
|
| 27 |
"--gpu-memory-utilization", "0.90",
|
| 28 |
"--trust-remote-code",
|
|
|
|
| 29 |
]
|
| 30 |
if "AWQ" in MODEL_ID.upper():
|
| 31 |
VLLM_ARGS += ["--quantization", "awq_marlin"] # faster AWQ kernel if available
|
|
|
|
| 26 |
"--max-model-len", "8192", # fits A10G 24GB
|
| 27 |
"--gpu-memory-utilization", "0.90",
|
| 28 |
"--trust-remote-code",
|
| 29 |
+
"--enable-auto-tool-choice", # Enable automatic tool selection
|
| 30 |
]
|
| 31 |
if "AWQ" in MODEL_ID.upper():
|
| 32 |
VLLM_ARGS += ["--quantization", "awq_marlin"] # faster AWQ kernel if available
|