From ad049c51c146e9b7852f87e2f040e97d88848a1f Mon Sep 17 00:00:00 2001 From: allegroai Date: Mon, 12 Apr 2021 03:27:21 +0300 Subject: [PATCH] typo --- clearml_serving/serving_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clearml_serving/serving_service.py b/clearml_serving/serving_service.py index f036d74..f7a46a7 100644 --- a/clearml_serving/serving_service.py +++ b/clearml_serving/serving_service.py @@ -52,7 +52,7 @@ class ServingService(object): self._task = Task.create( project_name=task_project, task_name=task_name, task_type=Task.TaskTypes.service, repo="https://github.com/allegroai/clearml-serving.git", - branch="master", + branch="main", script="clearml_serving/service.py", working_directory=".", add_task_init_call=False, @@ -189,7 +189,7 @@ class ServingService(object): task_name="triton serving engine", task_type=Task.TaskTypes.inference, repo="https://github.com/allegroai/clearml-serving.git", - branch="master", + branch="main", script="clearml_serving/triton_helper.py", working_directory=".", docker="nvcr.io/nvidia/tritonserver:21.03-py3 --ipc=host -p 8000:8000 -p 8001:8001 -p 8002:8002",