temporarily use pschroedl/ai-runner:lipsync for deployment

livepeer · Sep 19, 2024 · 002b37a · 002b37a
1 parent 74b4a73
commit 002b37a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 {