diff --git a/CHANGELOG_PENDING.md b/CHANGELOG_PENDING.md index 3c04750831..eb9267b8af 100644 --- a/CHANGELOG_PENDING.md +++ b/CHANGELOG_PENDING.md @@ -14,7 +14,8 @@ #### Orchestrator -* [#3814](https://github.com/livepeer/go-livepeer/pull/3814) Sai/worker: Add scope pipeline support to worker and build scripts (@victorges) +* [#3814](https://github.com/livepeer/go-livepeer/pull/3814) ai/worker: Add scope pipeline support to worker and build scripts (@victorges) +* [#3823](https://github.com/livepeer/go-livepeer/pull/3823) ai/worker: Add sd15-v2v image support (@victorges) #### Transcoder diff --git a/ai/worker/docker.go b/ai/worker/docker.go index 145188c269..b1fbd52003 100644 --- a/ai/worker/docker.go +++ b/ai/worker/docker.go @@ -72,6 +72,8 @@ var livePipelineToImage = map[string]string{ "streamdiffusion": "livepeer/ai-runner:live-app-streamdiffusion", // streamdiffusion-sd15 is a utility image that uses an SD1.5 model on the default config of the pipeline. Optimizes startup time. "streamdiffusion-sd15": "livepeer/ai-runner:live-app-streamdiffusion-sd15", + // streamdiffusion-sd15-v2v is a utility image that uses an SD1.5 model with cached attention enabled by default. Optimizes startup time for cached-attention inference. + "streamdiffusion-sd15-v2v": "livepeer/ai-runner:live-app-streamdiffusion-sd15-v2v", // streamdiffusion-sdxl is a utility image that uses an SDXL model on the default config of the pipeline. Optimizes startup time. "streamdiffusion-sdxl": "livepeer/ai-runner:live-app-streamdiffusion-sdxl", // streamdiffusion-sdxl-faceid is a utility image that uses an SDXL model with a FaceID IP Adapter on the default config of the pipeline. Optimizes startup time. diff --git a/ai/worker/docker_test.go b/ai/worker/docker_test.go index 6362065b01..2dc87fc070 100644 --- a/ai/worker/docker_test.go +++ b/ai/worker/docker_test.go @@ -334,6 +334,14 @@ func TestDockerManager_getContainerImageName(t *testing.T) { expectedImage: "livepeer/ai-runner:live-app-streamdiffusion", expectError: false, }, + { + name: "live-video-to-video with cached attention sd15 modelID", + setup: func(dockerManager *DockerManager, mockDockerClient *MockDockerClient) {}, + pipeline: "live-video-to-video", + modelID: "streamdiffusion-sd15-v2v", + expectedImage: "livepeer/ai-runner:live-app-streamdiffusion-sd15-v2v", + expectError: false, + }, { name: "live-video-to-video with invalid modelID", setup: func(dockerManager *DockerManager, mockDockerClient *MockDockerClient) {},