Update bhadresh-savani_distilbert-base-uncased-emotion_onnx_tokenize/1/model.py
Browse files
bhadresh-savani_distilbert-base-uncased-emotion_onnx_tokenize/1/model.py
CHANGED
@@ -42,7 +42,8 @@ class TritonPythonModel:
|
|
42 |
"""
|
43 |
# more variables in https://github.com/triton-inference-server/python_backend/blob/main/src/python.cc
|
44 |
|
45 |
-
path: str = str(Path(args["model_repository"]).
|
|
|
46 |
self.tokenizer = AutoTokenizer.from_pretrained(path)
|
47 |
|
48 |
def execute(self, requests) -> "List[List[pb_utils.Tensor]]":
|
|
|
42 |
"""
|
43 |
# more variables in https://github.com/triton-inference-server/python_backend/blob/main/src/python.cc
|
44 |
|
45 |
+
path: str = str(Path(args["model_repository"]).absolute()) + "/1/"
|
46 |
+
|
47 |
self.tokenizer = AutoTokenizer.from_pretrained(path)
|
48 |
|
49 |
def execute(self, requests) -> "List[List[pb_utils.Tensor]]":
|