Update model.py

This commit is contained in:
groupuser 2025-05-13 08:40:07 +00:00
parent a3dab09801
commit a91c637c60

@ -19,8 +19,8 @@ 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/gemma-3-1b-it/gemma-3-1b-it" #model_path = "/cheetah/input/model/gemma-3-1b-it/gemma-3-1b-it"
input_model_path = model_path input_model_path = model_path