diff --git a/1/model.py b/1/model.py index 2f2710b..d367f9d 100644 --- a/1/model.py +++ b/1/model.py @@ -20,7 +20,7 @@ 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 = model_repository self.logger.log_info(f"model_name: {self.model_name}") self.logger.log_info(f"model_repository: {model_repository}")