Bläddra i källkod

Python 3.11 (#1171)

* python 3.11

* python 3.11

* fix

* fix

* fix

* revert changes

* Update requirements.txt

* Trying pip3 install instead

* Excluding cp39 - torch 1.10.2

* Removing 1.10.2 from test

---------

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
Johnny 1 år sedan
förälder
incheckning
e69930cb9c
3 ändrade filer med 11 tillägg och 8 borttagningar
  1. 7 6
      .github/workflows/test.yml
  2. 3 1
      README.md
  3. 1 1
      requirements.txt

+ 7 - 6
.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?)$' .

+ 3 - 1
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
 

+ 1 - 1
requirements.txt

@@ -3,5 +3,5 @@ numpy
 torch
 tqdm
 more-itertools
-tiktoken==0.3.1
+tiktoken==0.3.3
 ffmpeg-python==0.2.0