groupuser/kicon_e5large_15_v1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
groupuser 2025-04-30 01:40:12 +00:00
parent ae17e8df35
commit 3fcea63de1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
1/model.py
View File

@ -4,8 +4,8 @@ import torch
class TritonPythonModel: class TritonPythonModel:
def initialize(self, args): def initialize(self, args):
self.tokenizer = AutoTokenizer.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 = AutoModel.from_pretrained("../../kicon_e5large_15_v1")
self.model.eval() self.model.eval()
def execute(self, requests): def execute(self, requests):