Skip to content

Commit c39c875

Browse files
Fix compose file and functional tests for Avatarchatbot on AMD ROCm platform (#1872)
Signed-off-by: Artem Astafev <[email protected]>
1 parent c2e9a25 commit c39c875

File tree

3 files changed

+5
-23
lines changed

3 files changed

+5
-23
lines changed

AvatarChatbot/docker_compose/amd/gpu/rocm/compose.yaml

Lines changed: 2 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ services:
4242
environment:
4343
TTS_ENDPOINT: ${TTS_ENDPOINT}
4444
tgi-service:
45-
image: ghcr.io/huggingface/text-generation-inference:2.3.1-rocm
45+
image: ghcr.io/huggingface/text-generation-inference:2.4.1-rocm
4646
container_name: tgi-service
4747
ports:
4848
- "${TGI_SERVICE_PORT:-3006}:80"
@@ -66,24 +66,6 @@ services:
6666
- seccomp:unconfined
6767
ipc: host
6868
command: --model-id ${LLM_MODEL_ID} --max-input-length 4096 --max-total-tokens 8192
69-
llm:
70-
image: ${REGISTRY:-opea}/llm-textgen:${TAG:-latest}
71-
container_name: llm-tgi-server
72-
depends_on:
73-
- tgi-service
74-
ports:
75-
- "3007:9000"
76-
ipc: host
77-
environment:
78-
no_proxy: ${no_proxy}
79-
http_proxy: ${http_proxy}
80-
https_proxy: ${https_proxy}
81-
TGI_LLM_ENDPOINT: ${TGI_LLM_ENDPOINT}
82-
LLM_ENDPOINT: ${TGI_LLM_ENDPOINT}
83-
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
84-
HF_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
85-
OPENAI_API_KEY: ${OPENAI_API_KEY}
86-
restart: unless-stopped
8769
wav2lip-service:
8870
image: ${REGISTRY:-opea}/wav2lip:${TAG:-latest}
8971
container_name: wav2lip-service
@@ -125,7 +107,7 @@ services:
125107
container_name: avatarchatbot-backend-server
126108
depends_on:
127109
- asr
128-
- llm
110+
- tgi-service
129111
- tts
130112
- animation
131113
ports:

AvatarChatbot/docker_compose/amd/gpu/rocm/set_env.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ export ANIMATION_SERVICE_HOST_IP=${host_ip}
3030
export MEGA_SERVICE_PORT=8888
3131
export ASR_SERVICE_PORT=3001
3232
export TTS_SERVICE_PORT=3002
33-
export LLM_SERVICE_PORT=3007
33+
export LLM_SERVICE_PORT=3006
3434
export ANIMATION_SERVICE_PORT=3008
3535

3636
export DEVICE="cpu"

AvatarChatbot/tests/test_compose_on_rocm.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ function build_docker_images() {
2727
git clone https://github.com/opea-project/GenAIComps.git && cd GenAIComps && git checkout "${opea_branch:-"main"}" && cd ../
2828

2929
echo "Build all the images with --no-cache, check docker_image_build.log for details..."
30-
service_list="avatarchatbot whisper asr llm-textgen speecht5 tts wav2lip animation"
30+
service_list="avatarchatbot whisper asr speecht5 tts wav2lip animation"
3131
docker compose -f build.yaml build ${service_list} --no-cache > ${LOG_PATH}/docker_image_build.log
3232

3333
docker pull ghcr.io/huggingface/text-generation-inference:2.3.1-rocm
@@ -65,7 +65,7 @@ function start_services() {
6565
export MEGA_SERVICE_PORT=8888
6666
export ASR_SERVICE_PORT=3001
6767
export TTS_SERVICE_PORT=3002
68-
export LLM_SERVICE_PORT=3007
68+
export LLM_SERVICE_PORT=3006
6969
export ANIMATION_SERVICE_PORT=3008
7070

7171
export DEVICE="cpu"

0 commit comments

Comments
 (0)