diff --git a/1/model.py b/1/model.py index bbd0d27..a0b8378 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/gemma-3-1b-it" + model_path = "/cheetah/input/model/gemma-3-1b-it/gemma-3-1b-it" input_model_path = model_path