diff --git a/1/model.py b/1/model.py index 8c7ade9..a58caed 100644 --- a/1/model.py +++ b/1/model.py @@ -19,8 +19,8 @@ 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 = f"{model_repository}/{self.model_name}" + model_path = "/cheetah/input/model/triton-gemma-3-1b-it" input_model_path = model_path