Update model.py

This commit is contained in:
groupuser 2025-05-21 02:27:03 +00:00
parent dd45caee91
commit ece50c848b

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