diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index bca49b5..3796a39 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -11,15 +11,16 @@ jobs: runs-on: ubuntu-latest strategy: matrix: - python-version: ['3.8', '3.9', '3.10'] - pytorch-version: [1.10.2, 1.13.1] + python-version: ['3.8', '3.9', '3.10', '3.11'] + pytorch-version: [1.13.1, 2.0.0] exclude: - - python-version: '3.10' - pytorch-version: 1.10.2 + - python-version: '3.11' + pytorch-version: 1.13.1 steps: - uses: conda-incubator/setup-miniconda@v2 - - run: conda install -n test ffmpeg python=${{ matrix.python-version }} pytorch=${{ matrix.pytorch-version }} cpuonly -c pytorch - - uses: actions/checkout@v2 + - run: conda install -n test ffmpeg python=${{ matrix.python-version }} + - run: pip3 install torch==${{ matrix.pytorch-version }}+cpu --index-url https://download.pytorch.org/whl/cpu + - uses: actions/checkout@v3 - run: echo "$CONDA/envs/test/bin" >> $GITHUB_PATH - run: pip install .["dev"] - run: black --check --diff -t py38 --include '(\.pyi?)$' . diff --git a/README.md b/README.md index 64e2d84..648d0c1 100644 --- a/README.md +++ b/README.md @@ -17,7 +17,9 @@ A Transformer sequence-to-sequence model is trained on various speech processing ## Setup -We used Python 3.9.9 and [PyTorch](https://pytorch.org/) 1.10.1 to train and test our models, but the codebase is expected to be compatible with Python 3.8-3.10 and recent PyTorch versions. The codebase also depends on a few Python packages, most notably [OpenAI's tiktoken](https://github.com/openai/tiktoken) for their fast tokenizer implementation and [ffmpeg-python](https://github.com/kkroening/ffmpeg-python) for reading audio files. You can download and install (or update to) the latest release of Whisper with the following command: + +We used Python 3.9.9 and [PyTorch](https://pytorch.org/) 1.10.1 to train and test our models, but the codebase is expected to be compatible with Python 3.8-3.11 and recent PyTorch versions. The codebase also depends on a few Python packages, most notably [OpenAI's tiktoken](https://github.com/openai/tiktoken) for their fast tokenizer implementation and [ffmpeg-python](https://github.com/kkroening/ffmpeg-python) for reading audio files. You can download and install (or update to) the latest release of Whisper with the following command: + pip install -U openai-whisper diff --git a/requirements.txt b/requirements.txt index 09ad0e3..995977a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,5 +3,5 @@ numpy torch tqdm more-itertools -tiktoken==0.3.1 +tiktoken==0.3.3 ffmpeg-python==0.2.0