test env

app.py

@@ -3,6 +3,60 @@
 # See the LICENSE file in the project root for full license information.
 
 import os
+
+from scripts.check_packages import (
+    install_flash_attn,
+    install_xformers,
+    install_kaolin,
+    install_nvdiffrast,
+    install_diffoctreerast,
+    install_vox2seq
+)
+
+# install flash-attn
+try:
+    import flash_attn
+except ModuleNotFoundError:
+    install_flash_attn()
+    import flash_attn
+
+# install xformers
+try:
+    import xformers
+    os.environ['ATTN_BACKEND'] = 'xformers'
+except ModuleNotFoundError:
+    install_xformers()
+    import xformers
+    os.environ['ATTN_BACKEND'] = 'xformers'
+
+# install kaolin
+try:
+    import kaolin
+except ModuleNotFoundError:
+    install_kaolin()
+    import kaolin
+
+# install nvdiffrast
+try:
+    import nvdiffrast
+except ModuleNotFoundError:
+    install_nvdiffrast()
+    import nvdiffrast
+
+# install diffoctreerast
+try:
+    import diffoctreerast
+except ModuleNotFoundError:
+    install_diffoctreerast()
+    import diffoctreerast
+
+# install vox2seq
+try:
+    import vox2seq
+except ModuleNotFoundError:
+    install_vox2seq()
+    import vox2seq
+
 os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "gradio_temp")
 os.makedirs(os.environ["GRADIO_TEMP_DIR"], exist_ok=True)
 import uuid

requirements.txt

@@ -30,3 +30,4 @@ pycocotools
 shapely
 gradio==4.44.1
 gradio_image_prompter
+spconv-cu118

scripts/check_packages.py

@@ -0,0 +1,48 @@
+import os
+
+def install_flash_attn():
+    print ("installing flash-attn...")
+    result = os.system('pip install flash-attn')
+    print (result)
+
+def install_xformers():
+    print ("installing xformers...")
+    result = os.system('pip install -U xformers --index-url https://download.pytorch.org/whl/cu129')
+    print (result)
+
+def install_kaolin():
+    print ("installing kaolin...")
+    result = os.system('pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.8.0_cu129.html')
+    print (result)
+
+def install_nvdiffrast():
+    print ("installing nvdiffrast...")
+    result = os.system('pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation')
+    print (result)
+
+def install_diffoctreerast():
+    print ("installing diffoctreerast...")
+    os.system('mkdir tmp')
+    os.system('mkdir tmp/extensions')
+    os.system('git clone --recurse-submodules https://github.com/JeffreyXiang/diffoctreerast.git ./tmp/extensions/diffoctreerast')
+    result = os.system('pip install ./tmp/extensions/diffoctreerast')
+    os.system('rm -rf ./tmp')
+    print (result)
+
+def install_mipsplatting():
+    print ("installing mipsplatting...")
+    os.system('mkdir tmp')
+    os.system('mkdir tmp/extensions')
+    os.system('git clone https://github.com/autonomousvision/mip-splatting.git ./tmp/extensions/mip-splatting')
+    result = os.system('pip install ./tmp/extensions/mip-splatting/submodules/diff-gaussian-rasterization/')
+    os.system('rm -rf ./tmp')
+    print (result)
+
+def install_vox2seq():
+    print ("installing vox2seq...")
+    os.system('mkdir tmp')
+    os.system('mkdir tmp/extensions')
+    os.system('git clone https://github.com/microsoft/TRELLIS.git@334d3b2fa1212477230ea1be989447273f389618 ./tmp/extensions/TRELLIS')
+    result = os.system('pip install ./tmp/extensions/TRELLIS/extensions/vox2seq')
+    os.system('rm -rf ./tmp')
+    print (result)