From 3fcea63de1cffcb041989a875ee75e987deaf016 Mon Sep 17 00:00:00 2001 From: groupuser Date: Wed, 30 Apr 2025 01:40:12 +0000 Subject: [PATCH] Update model.py --- 1/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/1/model.py b/1/model.py index 167e5cf..9016e59 100644 --- a/1/model.py +++ b/1/model.py @@ -4,8 +4,8 @@ import torch class TritonPythonModel: def initialize(self, args): - self.tokenizer = AutoTokenizer.from_pretrained("../kicon_e5large_15_v1") - self.model = AutoModel.from_pretrained("../kicon_e5large_15_v1") + self.tokenizer = AutoTokenizer.from_pretrained("../../kicon_e5large_15_v1") + self.model = AutoModel.from_pretrained("../../kicon_e5large_15_v1") self.model.eval() def execute(self, requests):