Peter Larnholt commited on
Commit
55381a2
·
1 Parent(s): 1b713bd

Enable auto tool choice for better tool calling support

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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