groupuser/triton-gemma-3-1b-it
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-05-20 06:05:23 +00:00
parent f7f4d8c5f2
commit 35b84ab231
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

@ -19,8 +19,8 @@ class TritonPythonModel:
self.model_name = args["model_name"]
model_repository = args["model_repository"]
model_path = f"{model_repository}/{self.model_name}"
#model_path = "/cheetah/input/model/triton-gemma-3-1b-it"
#model_path = f"{model_repository}/{self.model_name}"
model_path = "/cheetah/input/model/triton-gemma-3-1b-it"
input_model_path = model_path