Skip to content

Commit

Permalink
Update base image for tts asr import check test (NVIDIA#10072)
Browse files Browse the repository at this point in the history
* Fix torch version for tts asr import check test

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Ignore torch requirement

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

* Update base image used for import check

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

---------

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
Co-authored-by: Dong Hyuk Chang <donghyukc@nvidia.com>
  • Loading branch information
thomasdhc and Dong Hyuk Chang authored Aug 8, 2024
1 parent e879330 commit 3ba23bd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/import-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ jobs:
test-asr-imports:
runs-on: ubuntu-latest
container:
image: pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
steps:
- name: Checkout repo
uses: actions/checkout@v2
Expand Down Expand Up @@ -43,7 +43,7 @@ jobs:
test-tts-imports:
runs-on: ubuntu-latest
container:
image: pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
steps:
- name: Checkout repo
uses: actions/checkout@v2
Expand All @@ -70,4 +70,4 @@ jobs:
# Run import checks
python tests/core_ptl/check_imports.py --domain "tts"
# Uninstall NeMo
pip uninstall -y nemo_toolkit
pip uninstall -y nemo_toolkit

0 comments on commit 3ba23bd

Please sign in to comment.