Update setup.py
Browse files
setup.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
from typing import List
|
| 2 |
-
|
| 3 |
from setuptools import find_packages, setup
|
| 4 |
-
|
| 5 |
|
| 6 |
def fetch_requirements(path) -> List[str]:
|
| 7 |
"""
|
|
@@ -16,7 +15,6 @@ def fetch_requirements(path) -> List[str]:
|
|
| 16 |
with open(path, "r") as fd:
|
| 17 |
return [r.strip() for r in fd.readlines()]
|
| 18 |
|
| 19 |
-
|
| 20 |
def fetch_readme() -> str:
|
| 21 |
"""
|
| 22 |
This function reads the README.md file in the current directory.
|
|
@@ -27,7 +25,6 @@ def fetch_readme() -> str:
|
|
| 27 |
with open("README.md", encoding="utf-8") as f:
|
| 28 |
return f.read()
|
| 29 |
|
| 30 |
-
|
| 31 |
setup(
|
| 32 |
name="opensora",
|
| 33 |
version="1.0.0",
|
|
@@ -58,3 +55,17 @@ setup(
|
|
| 58 |
"Topic :: System :: Distributed Computing",
|
| 59 |
],
|
| 60 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
from typing import List
|
|
|
|
| 2 |
from setuptools import find_packages, setup
|
| 3 |
+
import subprocess
|
| 4 |
|
| 5 |
def fetch_requirements(path) -> List[str]:
|
| 6 |
"""
|
|
|
|
| 15 |
with open(path, "r") as fd:
|
| 16 |
return [r.strip() for r in fd.readlines()]
|
| 17 |
|
|
|
|
| 18 |
def fetch_readme() -> str:
|
| 19 |
"""
|
| 20 |
This function reads the README.md file in the current directory.
|
|
|
|
| 25 |
with open("README.md", encoding="utf-8") as f:
|
| 26 |
return f.read()
|
| 27 |
|
|
|
|
| 28 |
setup(
|
| 29 |
name="opensora",
|
| 30 |
version="1.0.0",
|
|
|
|
| 55 |
"Topic :: System :: Distributed Computing",
|
| 56 |
],
|
| 57 |
)
|
| 58 |
+
|
| 59 |
+
install_options = [
|
| 60 |
+
"--disable-pip-version-check",
|
| 61 |
+
"--no-cache-dir",
|
| 62 |
+
"--no-build-isolation",
|
| 63 |
+
"--config-settings", "--build-option=--cpp_ext",
|
| 64 |
+
"--config-settings", "--build-option=--cuda_ext"
|
| 65 |
+
]
|
| 66 |
+
|
| 67 |
+
subprocess.run(
|
| 68 |
+
["pip", "install", "-v"] + install_options + ["git+https://github.com/sandeshrajbhandari/apex.git"],
|
| 69 |
+
check=True,
|
| 70 |
+
capture_output=True
|
| 71 |
+
)
|