Skip to content

Commit ff6536c

Browse files
authored
Merge pull request #41 from EmbeddedLLM/szeyu-patch-4
fix the og.Model(model_path) to use self.model_path instead
2 parents b57366e + bba6091 commit ff6536c

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)