Open-Sora / setup.py
kadirnar's picture
Update setup.py
ca7ce2f verified
raw
history blame
1.93 kB
from typing import List
from setuptools import find_packages, setup
import subprocess
def fetch_requirements(path) -> List[str]:
"""
This function reads the requirements file.
Args:
path (str): the path to the requirements file.
Returns:
The lines in the requirements file.
"""
with open(path, "r") as fd:
return [r.strip() for r in fd.readlines()]
def fetch_readme() -> str:
"""
This function reads the README.md file in the current directory.
Returns:
The lines in the README file.
"""
with open("README.md", encoding="utf-8") as f:
return f.read()
setup(
name="opensora",
version="1.0.0",
packages=find_packages(
exclude=(
"assets",
"configs",
"docs",
"outputs",
"pretrained_models",
"scripts",
"tests",
"tools",
"*.egg-info",
)
),
description="Democratizing Efficient Video Production for All",
long_description=fetch_readme(),
long_description_content_type="text/markdown",
license="Apache Software License 2.0",
install_requires=fetch_requirements("requirements.txt"),
python_requires=">=3.6",
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Environment :: GPU :: NVIDIA CUDA",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: System :: Distributed Computing",
],
)
install_options = [
"--disable-pip-version-check",
"--no-cache-dir",
"--no-build-isolation",
"--config-settings", "--build-option=--cpp_ext",
"--config-settings", "--build-option=--cuda_ext"
]
subprocess.run(
["pip", "install", "-v"] + install_options + ["git+https://github.com/sandeshrajbhandari/apex.git"],
check=True,
capture_output=True
)