From 7eb360d0d5f3eb03292d3ff4596a8d50c9765888 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sun, 18 Feb 2024 20:32:40 +0800 Subject: [PATCH] Fix cpu docker images for torch 2.2.0 (#1502) --- .github/scripts/docker/generate_build_matrix.py | 4 ++-- .github/workflows/yesno.yml | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/scripts/docker/generate_build_matrix.py b/.github/scripts/docker/generate_build_matrix.py index f0690f8bf0..425afac2b9 100755 --- a/.github/scripts/docker/generate_build_matrix.py +++ b/.github/scripts/docker/generate_build_matrix.py @@ -43,8 +43,8 @@ def get_torchaudio_version(torch_version): def get_matrix(): - k2_version = "1.24.4.dev20240211" - kaldifeat_version = "1.25.4.dev20240210" + k2_version = "1.24.4.dev20240218" + kaldifeat_version = "1.25.4.dev20240218" version = "1.3" python_version = ["3.8", "3.9", "3.10", "3.11", "3.12"] torch_version = ["1.13.0", "1.13.1", "2.0.0", "2.0.1", "2.1.0", "2.1.1", "2.1.2"] diff --git a/.github/workflows/yesno.yml b/.github/workflows/yesno.yml index 182300dfa1..de822b33f6 100644 --- a/.github/workflows/yesno.yml +++ b/.github/workflows/yesno.yml @@ -59,4 +59,7 @@ jobs: cd /icefall git config --global --add safe.directory /icefall + python3 -m torch.utils.collect_env + python3 -m k2.version + .github/scripts/yesno/ASR/run.sh