x
This commit is contained in:
@@ -144,6 +144,11 @@ def launch_model(model_id, config, model_path, gpu_count):
|
||||
"--gpu-memory-utilization", str(gpu_util),
|
||||
"--dtype", "auto"
|
||||
]
|
||||
|
||||
model_impl = config.get("model_impl") or os.getenv("VLLM_MODEL_IMPL")
|
||||
if model_impl:
|
||||
cmd.extend(["--model-impl", str(model_impl)])
|
||||
log(f"Added model implementation: {model_impl}")
|
||||
|
||||
if config.get("trust_remote"):
|
||||
cmd.append("--trust-remote-code")
|
||||
|
||||
Reference in New Issue
Block a user