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:23:30 +00:00
parent bbe8d5934f
commit da9dfb3779
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
1/model.py
View File

@ -20,7 +20,7 @@ class TritonPythonModel:
self.model_name = args["model_name"] self.model_name = args["model_name"]
model_repository = args["model_repository"] model_repository = args["model_repository"]
#model_path = f"{model_repository}/{self.model_name}" #model_path = f"{model_repository}/{self.model_name}"
model_path = "/cheetah/input/model/triton-gemma-3-1b-it" model_path = model_repository
self.logger.log_info(f"model_name: {self.model_name}") self.logger.log_info(f"model_name: {self.model_name}")
self.logger.log_info(f"model_repository: {model_repository}") self.logger.log_info(f"model_repository: {model_repository}")