Skip to content

Commit 3db7e4b

Browse files
committed
Resolve loading loop for local models
1 parent 8fd0fd0 commit 3db7e4b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

interpreter/core/llm/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -306,6 +306,8 @@ def load(self):
306306
if self._is_loaded:
307307
return
308308

309+
self._is_loaded = True
310+
309311
if self.model.startswith("ollama/"):
310312
model_name = self.model.replace("ollama/", "")
311313
try:
@@ -371,8 +373,6 @@ def load(self):
371373
except:
372374
pass
373375

374-
self._is_loaded = True
375-
376376

377377
def fixed_litellm_completions(**params):
378378
"""

0 commit comments

Comments
 (0)