Skip to content

Commit 30a88bd

Browse files
fix: pass revision to model loader in mlx backend
1 parent d7a6c20 commit 30a88bd

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/infer_check/backends/mlx_lm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def _ensure_loaded(self) -> None:
121121

122122
repo_or_path = str(model_path) if model_path.exists() else self._model_id
123123
try:
124-
res = load(repo_or_path)
124+
res = load(repo_or_path, revision=self._revision)
125125
except Exception as exc:
126126
msg = str(exc)
127127
if "404" in msg or "Repository Not Found" in msg:

0 commit comments

Comments
 (0)