Skip to content

Commit

Permalink
mosnet seems incompatible with numpy>=1.24. updated the setup and readme
Browse files Browse the repository at this point in the history
  • Loading branch information
aliutkus committed Jul 5, 2023
1 parent 12f6fd2 commit 74619e9
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 15 deletions.
13 changes: 3 additions & 10 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,18 +10,11 @@ If you find speechmetrics useful, you are welcome to cite the original papers fo
# Installation
As of our recent tests, installation goes smoothly on ubuntu, but there may be some compiler errors for `pypesq` on iOs.

Note that the mosnet seems to be incompatible with numpy >= 1.24

For cpu usage:
```
pip install numpy
pip install git+https://github.com/aliutkus/speechmetrics#egg=speechmetrics[cpu]
```

For gpu usage (on the MOSNet)

```
pip install numpy
pip install git+https://github.com/aliutkus/speechmetrics#egg=speechmetrics[gpu]
pip install numpy==1.23.4
pip install git+https://github.com/aliutkus/speechmetrics#egg=speechmetrics
```

# Usage
Expand Down
8 changes: 3 additions & 5 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,14 @@
packages=find_packages(),

install_requires=[
'numpy',
'numpy<1.24',
'scipy',
'tqdm',
'resampy',
'pystoi',
'museval',
'tensorflow>=2.0.0',
'librosa',
# This is requred, but srmrpy pull it in,
# and there is a pip3 conflict if we have the following
# line.
Expand All @@ -24,9 +26,5 @@
'srmrpy @ git+https://github.com/jfsantos/SRMRpy',
'pesq @ git+https://github.com/ludlows/python-pesq',
],
extras_require={
'cpu': ['tensorflow>=2.0.0', 'librosa'],
'gpu': ['tensorflow>=2.0.0', 'librosa'],
},
include_package_data=True
)

0 comments on commit 74619e9

Please sign in to comment.