Minor fix after df7bb0db1f

This commit is contained in:
oobabooga 2025-05-05 04:54:25 -07:00
parent f3da45f65d
commit b817bb33fd
2 changed files with 2 additions and 2 deletions

View file

@ -256,7 +256,7 @@ class LlamaServer:
self.server_path,
"--model", self.model_path,
"--ctx-size", str(shared.args.ctx_size),
"--n-gpu-layers", str(shared.args.n_gpu_layers),
"--gpu-layers", str(shared.args.gpu_layers),
"--batch-size", str(shared.args.batch_size),
"--port", str(self.port),
]