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):