Update setup.py

setup.py

@@ -1,7 +1,6 @@
 from typing import List
-
 from setuptools import find_packages, setup
-
+import subprocess
 
 def fetch_requirements(path) -> List[str]:
     """
@@ -16,7 +15,6 @@ def fetch_requirements(path) -> List[str]:
     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.
@@ -27,7 +25,6 @@ def fetch_readme() -> str:
     with open("README.md", encoding="utf-8") as f:
         return f.read()
 
-
 setup(
     name="opensora",
     version="1.0.0",
@@ -58,3 +55,17 @@ setup(
         "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
+)