This commit is contained in:
allegroai 2021-04-12 03:27:21 +03:00
parent d980ee3519
commit ad049c51c1

View File

@ -52,7 +52,7 @@ class ServingService(object):
self._task = Task.create( self._task = Task.create(
project_name=task_project, task_name=task_name, task_type=Task.TaskTypes.service, project_name=task_project, task_name=task_name, task_type=Task.TaskTypes.service,
repo="https://github.com/allegroai/clearml-serving.git", repo="https://github.com/allegroai/clearml-serving.git",
branch="master", branch="main",
script="clearml_serving/service.py", script="clearml_serving/service.py",
working_directory=".", working_directory=".",
add_task_init_call=False, add_task_init_call=False,
@ -189,7 +189,7 @@ class ServingService(object):
task_name="triton serving engine", task_name="triton serving engine",
task_type=Task.TaskTypes.inference, task_type=Task.TaskTypes.inference,
repo="https://github.com/allegroai/clearml-serving.git", repo="https://github.com/allegroai/clearml-serving.git",
branch="master", branch="main",
script="clearml_serving/triton_helper.py", script="clearml_serving/triton_helper.py",
working_directory=".", working_directory=".",
docker="nvcr.io/nvidia/tritonserver:21.03-py3 --ipc=host -p 8000:8000 -p 8001:8001 -p 8002:8002", docker="nvcr.io/nvidia/tritonserver:21.03-py3 --ipc=host -p 8000:8000 -p 8001:8001 -p 8002:8002",