Skip to content

Commit bba6091

Browse files
committed
fix the og.Model(model_path) to use self.model_path instead
1 parent 3ad496d commit bba6091

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/embeddedllm/backend/onnxruntime_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def __init__(self, model_path: str, vision: bool, device: str = "cpu"):
7070
logger.info("Attempt to load slower tokenizer")
7171
self.tokenizer = PreTrainedTokenizer.from_pretrained(self.model_path)
7272

73-
self.model = og.Model(model_path)
73+
self.model = og.Model(self.model_path)
7474
logger.info("Model loaded")
7575
self.onnx_tokenizer = og.Tokenizer(self.model)
7676
self.onnx_tokenizer_stream = self.onnx_tokenizer.create_stream()

0 commit comments

Comments
 (0)