Update model.py

This commit is contained in:
groupuser 2025-05-21 02:15:17 +00:00
parent 32e72102d2
commit dd45caee91

@ -19,7 +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 = model_repository #model_path = model_repository
model_path = "/cheetah/input/model/groupuser/base-gemma-3-1b-it/base-gemma-3-1b-it"
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}")