github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update Whisper to use fetch helper (#2401)

Browse Source 
* update whisper to use new fetch helper

* simplify file opening

* update name

* update key name to "downloads-cache"
This commit is contained in:
Francis Lata
2023-11-23 15:59:59 -05:00
committed by GitHub
parent 0505c5ea50
commit 6d672785db
2 changed files with 15 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
.github/workflows/test.yml vendored
View File

@@ -81,13 +81,11 @@ jobs:
with:
path: ${{ env.Python3_ROOT_DIR }}/lib/python3.8/site-packages
key: testing-packages-${{ hashFiles('**/setup.py') }}
- name: Cache model weights
- name: Cache downloads
uses: actions/cache@v3
with:
path: |
weights/whisper-tiny.en.pt
weights/gpt2.tiktoken
key: model-weights-v1
path: ~/.cache/tinygrad/downloads/
key: downloads-cache
- name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Pytest
@@ -116,13 +114,11 @@ jobs:
with:
path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages
key: testing-packages-${{ hashFiles('**/setup.py') }}
- name: Cache model weights
- name: Cache downloads
uses: actions/cache@v3
with:
path: |
weights/whisper-tiny.en.pt
weights/gpt2.tiktoken
key: model-weights-v1
path: ~/.cache/tinygrad/downloads/
key: downloads-cache
- name: Install Dependencies
run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Pytest
@@ -229,13 +225,11 @@ jobs:
key: metal-webgpu-testing-packages-${{ hashFiles('**/setup.py') }}
- name: Install Dependencies
run: pip install -e '.[webgpu,testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Cache model weights
- name: Cache downloads
uses: actions/cache@v3
with:
path: |
weights/whisper-tiny.en.pt
weights/gpt2.tiktoken
key: model-weights-v1
path: ~/Library/Caches/tinygrad/downloads/
key: downloads-cache
- name: Test LLaMA compile speed
run: PYTHONPATH="." METAL=1 python test/external/external_test_speed_llama.py
#- name: Run dtype test
@@ -296,13 +290,11 @@ jobs:
with:
path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages
key: ${{ matrix.backend }}-packages-${{ hashFiles('**/setup.py') }}
- name: Cache model weights
- name: Cache downloads
uses: actions/cache@v3
with:
path: |
weights/whisper-tiny.en.pt
weights/gpt2.tiktoken
key: model-weights-v1
path: ~/Library/Caches/tinygrad/downloads/
key: downloads-cache
- name: Set env
run: printf "${{ matrix.backend == 'llvm' && 'LLVM=1' || matrix.backend == 'clang' && 'CLANG=1' || matrix.backend == 'gpu' && 'GPU=1' || matrix.backend == 'cuda' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\n' || matrix.backend == 'PTX' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\nPTX=1' || matrix.backend == 'triton' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\nTRITON=1\nTRITON_PTXAS_PATH=/usr/bin/ptxas'}}" >> $GITHUB_ENV
- name: Install OpenCL