Skip to content

Commit

Permalink
Merge branch 'dev'
Browse files Browse the repository at this point in the history
  • Loading branch information
ControlNet committed Jan 28, 2024
2 parents 2547c63 + 420a3c0 commit 59b78ef
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 2 deletions.
3 changes: 3 additions & 0 deletions .github/workflows/unittest.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ jobs:
torch-version: "1.9.*"
- python-version: "3.10"
torch-version: "1.10.*"

- python-version: "3.11"
torch-version: "1.8.*"
- python-version: "3.11"
Expand All @@ -64,6 +65,8 @@ jobs:
torch-version: "1.11.*"
- python-version: "3.11"
torch-version: "1.12.*"
- python-version: "3.11"
torch-version: "1.13.*"

runs-on: ubuntu-20.04

Expand Down
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ The repository contains 2 parts:
## Use `marlin-pytorch` for Feature Extraction

Requirements:
- Python >= 3.6, < 3.11
- Python >= 3.6, < 3.12
- PyTorch >= 1.8
- ffmpeg

Expand Down Expand Up @@ -150,7 +150,7 @@ features = model.extract_features(x, keep_seq=False) # torch.Size([B, 768])
## Paper Implementation

### Requirements
- Python >= 3.7, < 3.11
- Python >= 3.7, < 3.12
- PyTorch ~= 1.11
- Torchvision ~= 0.12

Expand Down
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"License :: Other/Proprietary License",
"Operating System :: OS Independent",
"Intended Audience :: Developers",
Expand Down

0 comments on commit 59b78ef

Please sign in to comment.