mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

Do not install DeepSpeed under Windows (to-do: default backend to use local if on Windows)

Browse Source
This commit is contained in:
mrq 2023-08-24 14:27:36 -05:00
parent 22904a8639
commit b6c9686f7d
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@ -1,5 +1,5 @@
import subprocess import subprocess
import sys
from pathlib import Path from pathlib import Path
from datetime import datetime from datetime import datetime
from setuptools import setup, find_packages from setuptools import setup, find_packages
@ -37,9 +37,8 @@ setup(
long_description=long_description, long_description=long_description,
long_description_content_type="text/markdown", long_description_content_type="text/markdown",
packages=find_packages(), packages=find_packages(),
install_requires=[ install_requires=(["deepspeed>=0.7.7"] if not sys.platform.startswith("win") else []) +[
"coloredlogs>=15.0.1", "coloredlogs>=15.0.1",
"deepspeed>=0.7.7",
"diskcache>=5.4.0", "diskcache>=5.4.0",
"einops>=0.6.0", "einops>=0.6.0",
"encodec>=0.1.1", "encodec>=0.1.1",