Skip to content

Commit

Permalink
Install pytorch first to pin version
Browse files Browse the repository at this point in the history
  • Loading branch information
rockerBOO committed Nov 29, 2024
1 parent e59e276 commit dd3b846
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ jobs:
uses: actions/setup-python@v5
with:
python-version: '3.x'

- name: Install dependencies
run: python -m pip install --upgrade pip setuptools wheel

Expand All @@ -27,11 +28,13 @@ jobs:
with:
python-version: '3.x'
cache: 'pip' # caching pip dependencies

- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install dadaptation==3.2 torch==2.4.0 torchvision==0.19.0 accelerate==0.33.0
pip install -r requirements.txt
- name: Test with pytest
run: |
pip install pytest
Expand Down

0 comments on commit dd3b846

Please sign in to comment.