Skip to content

Commit 4e294bb

Browse files
committed
Extraction bug
1 parent 87964fc commit 4e294bb

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

squeez/inference/extractor.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,9 @@ def _init_vllm(self, base_url: str, model_name: str | None):
247247
self._backend = "vllm"
248248

249249
# Auto-detect model name from server if not provided
250-
if not model_name:
250+
if model_name:
251+
self._model_name = model_name
252+
else:
251253
try:
252254
models = self._client.models.list()
253255
self._model_name = models.data[0].id

0 commit comments

Comments
 (0)