diff --git a/worker/docker.go b/worker/docker.go index 3f4b529b..ac0aef26 100644 --- a/worker/docker.go +++ b/worker/docker.go @@ -21,8 +21,8 @@ import ( const containerModelDir = "/models" const containerPort = "8000/tcp" const pollingInterval = 500 * time.Millisecond -const containerTimeout = 2 * time.Minute -const externalContainerTimeout = 2 * time.Minute +const containerTimeout = 3 * time.Minute +const externalContainerTimeout = 3 * time.Minute const optFlagsContainerTimeout = 5 * time.Minute const containerRemoveTimeout = 30 * time.Second const containerCreatorLabel = "creator" @@ -43,7 +43,7 @@ var containerHostPorts = map[string]string{ // Mapping for per pipeline container images. var pipelineToImage = map[string]string{ "segment-anything-2": "livepeer/ai-runner:segment-anything-2", - "lipsync": "livepeer/ai-runner:lipsync", + "lipsync": "pschroedl/ai-runner:lipsync", } type DockerManager struct {