Skip to content

Commit a88a41c

Browse files
authored
ai/worker: Add scope pipeline support to worker and build scripts (#3814)
* Add scope pipeline support to worker and build scripts This commit adds support for the scope pipeline in the go-livepeer worker: Worker Integration: - Updated ai/worker/docker.go to map scope modelID to livepeer/ai-runner:live-app-scope image in livePipelineToImage map - Added test case in ai/worker/docker_test.go to verify scope image mapping works correctly Build Scripts: - Updated box/build-runner.sh to accept scope as a valid pipeline option alongside streamdiffusion and comfyui These changes enable the orchestrator to recognize and schedule scope pipeline containers for live-video-to-video requests. * Update CHANGELOG
1 parent 81f12af commit a88a41c

File tree

4 files changed

+14
-2
lines changed

4 files changed

+14
-2
lines changed

CHANGELOG_PENDING.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414

1515
#### Orchestrator
1616

17+
* [#3814](https://github.com/livepeer/go-livepeer/pull/3814) Sai/worker: Add scope pipeline support to worker and build scripts (@victorges)
18+
1719
#### Transcoder
1820

1921
### Bug Fixes 🐞

ai/worker/docker.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ var livePipelineToImage = map[string]string{
7777
// 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.
7878
"streamdiffusion-sdxl-faceid": "livepeer/ai-runner:live-app-streamdiffusion-sdxl-faceid",
7979
"comfyui": "livepeer/ai-runner:live-app-comfyui",
80+
"scope": "livepeer/ai-runner:live-app-scope",
8081
"segment_anything_2": "livepeer/ai-runner:live-app-segment_anything_2",
8182
"noop": "livepeer/ai-runner:live-app-noop",
8283
}

ai/worker/docker_test.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -433,6 +433,15 @@ func TestDockerManager_getContainerImageName(t *testing.T) {
433433
expectedImage: "livepeer/ai-runner:live-app-comfyui",
434434
expectError: false,
435435
},
436+
{
437+
name: "scope live image",
438+
setup: func(dockerManager *DockerManager, mockDockerClient *MockDockerClient) {
439+
},
440+
pipeline: "live-video-to-video",
441+
modelID: "scope",
442+
expectedImage: "livepeer/ai-runner:live-app-scope",
443+
expectError: false,
444+
},
436445
}
437446

438447
for _, tt := range tests {

box/build-runner.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ set -ex
33

44
PIPELINE=${PIPELINE:-noop}
55

6-
if [[ "$PIPELINE" != "noop" && "$PIPELINE" != "comfyui" && "$PIPELINE" != "streamdiffusion" && ! "$PIPELINE" =~ ^streamdiffusion- ]]; then
7-
echo "Error: PIPELINE must be either 'noop', 'comfyui', 'streamdiffusion' or start with 'streamdiffusion-'"
6+
if [[ "$PIPELINE" != "noop" && "$PIPELINE" != "comfyui" && "$PIPELINE" != "streamdiffusion" && "$PIPELINE" != "scope" && ! "$PIPELINE" =~ ^streamdiffusion- ]]; then
7+
echo "Error: PIPELINE must be either 'noop', 'comfyui', 'streamdiffusion', 'scope' or start with 'streamdiffusion-'"
88
exit 1
99
fi
1010

0 commit comments

Comments
 (0)