From 0979a3eb0f64ef76b13d753324a52e687e5af88e Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 22 Dec 2023 23:36:54 +0800
Subject: [PATCH] install ffmpeg for torch >= 2.1.0

---
 .github/scripts/docker/Dockerfile       | 1 +
 .github/workflows/build-cpu-docker.yml  | 2 +-
 .github/workflows/train-librispeech.yml | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/scripts/docker/Dockerfile b/.github/scripts/docker/Dockerfile
index 47af9983b..6f57c1ccb 100644
--- a/.github/scripts/docker/Dockerfile
+++ b/.github/scripts/docker/Dockerfile
@@ -12,6 +12,7 @@ ARG _KALDIFEAT_VERSION="${KALDIFEAT_VERSION}+cpu.torch${TORCH_VERSION}"
 
 RUN apt-get update -y && \
     apt-get install -qq -y \
+    ffmpeg \
     git \
     git-lfs \
     less \
diff --git a/.github/workflows/build-cpu-docker.yml b/.github/workflows/build-cpu-docker.yml
index bf1125779..c89655552 100644
--- a/.github/workflows/build-cpu-docker.yml
+++ b/.github/workflows/build-cpu-docker.yml
@@ -2,7 +2,7 @@ name: build-cpu-docker
 on:
   push:
     branches:
-      - ci-train-3
+      - ci-train-2
   workflow_dispatch:
 
 concurrency:
diff --git a/.github/workflows/train-librispeech.yml b/.github/workflows/train-librispeech.yml
index 0a76e0cd7..e9850bf84 100644
--- a/.github/workflows/train-librispeech.yml
+++ b/.github/workflows/train-librispeech.yml
@@ -3,7 +3,7 @@ on:
   push:
     branches:
       - master
-      - ci-train-2
+      - ci-train-3
 
   pull_request:
     branches: