Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update setup.py install tf with external pip call

This commit is contained in:
erogol 2020-07-12 15:36:36 +02:00
parent 238cc424f4
commit 0ab73061a1
1 changed files with 53 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
setup.py
View File

@ -69,6 +69,41 @@ if 'bdist_wheel' in unknown_args and args.checkpoint and args.model_config:
shutil.copy(args.model_config, embedded_config_path) shutil.copy(args.model_config, embedded_config_path)
package_data.extend([embedded_checkpoint_path, embedded_config_path]) package_data.extend([embedded_checkpoint_path, embedded_config_path])
def pip_install(package_name):
subprocess.call(
[sys.executable, '-m', 'pip', 'install', package_name]
)
requirements = {
'install_requires':[
"torch>=1.5",
"numpy>=1.16.0",
"numba==0.48",
"scipy>=0.19.0",
"librosa==0.6.2",
"unidecode==0.4.20",
"attrdict",
"tensorboardX",
"matplotlib",
"Pillow",
"flask",
"tqdm",
"inflect",
"bokeh==1.4.0",
"soundfile",
"phonemizer>=2.2.0",
"nose==1.3.7",
"cardboardlint==1.3.0",
"pylint==2.5.3",
],
'pip_install':[
'tensorflow>=2.2.0',
]
}
setup( setup(
name='TTS', name='TTS',
version=version, version=version,
@ -95,24 +130,23 @@ setup(
'build_py': build_py, 'build_py': build_py,
'develop': develop, 'develop': develop,
}, },
install_requires=[ install_requires=requirements['install_requires'],
"scipy>=0.19.0", python_requires='>=3.6.0',
"torch>=1.5", classifiers=[
"numpy>=1.16.0", "Programming Language :: Python",
"librosa==0.6.2", "Programming Language :: Python :: 3",
"unidecode==0.4.20", "Programming Language :: Python :: 3.6",
"attrdict", "Programming Language :: Python :: 3.7",
"tensorboardX", "Programming Language :: Python :: 3.8",
"matplotlib", 'Development Status :: 3 - Alpha',
"Pillow", "Intended Audience :: Science/Research :: Developers",
"flask", "Operating System :: POSIX :: Linux",
"tqdm", 'License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)',
"inflect", "Topic :: Software Development :: Libraries :: Python Modules :: Speech :: Sound/Audio :: Multimedia :: Artificial Intelligence",
"bokeh==1.4.0",
"soundfile",
"phonemizer @ https://github.com/bootphon/phonemizer/tarball/master",
],
dependency_links=[
"http://github.com/bootphon/phonemizer/tarball/master#egg=phonemizer-1.0.1"
] ]
) )
# for some reason having tensorflow in 'install_requires'
# breaks some of the dependencies.
for module in requirements['pip_install']:
pip_install(module)