Spaces:
Running on Zero
Running on Zero
Commit ·
e7a1c55
1
Parent(s): d8c9867
test py312
Browse files- README.md +1 -1
- app.py +4 -22
- scripts/check_packages.py +0 -9
README.md
CHANGED
|
@@ -5,7 +5,7 @@ colorFrom: blue
|
|
| 5 |
colorTo: indigo
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 4.44.1
|
| 8 |
-
python_version: '3.
|
| 9 |
app_file: app.py
|
| 10 |
pinned: false
|
| 11 |
license: apache-2.0
|
|
|
|
| 5 |
colorTo: indigo
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 4.44.1
|
| 8 |
+
python_version: '3.12'
|
| 9 |
app_file: app.py
|
| 10 |
pinned: false
|
| 11 |
license: apache-2.0
|
app.py
CHANGED
|
@@ -6,10 +6,8 @@ import os
|
|
| 6 |
|
| 7 |
from scripts.check_packages import (
|
| 8 |
install_flash_attn,
|
| 9 |
-
install_xformers,
|
| 10 |
install_kaolin,
|
| 11 |
install_nvdiffrast,
|
| 12 |
-
install_diffoctreerast,
|
| 13 |
install_vox2seq
|
| 14 |
)
|
| 15 |
|
|
@@ -18,44 +16,28 @@ try:
|
|
| 18 |
import flash_attn
|
| 19 |
except ModuleNotFoundError:
|
| 20 |
install_flash_attn()
|
| 21 |
-
import flash_attn
|
| 22 |
-
|
| 23 |
-
# install xformers
|
| 24 |
-
try:
|
| 25 |
-
import xformers
|
| 26 |
-
os.environ['ATTN_BACKEND'] = 'xformers'
|
| 27 |
-
except ModuleNotFoundError:
|
| 28 |
-
install_xformers()
|
| 29 |
-
import xformers
|
| 30 |
-
os.environ['ATTN_BACKEND'] = 'xformers'
|
| 31 |
|
| 32 |
# install kaolin
|
| 33 |
try:
|
| 34 |
import kaolin
|
| 35 |
except ModuleNotFoundError:
|
| 36 |
install_kaolin()
|
| 37 |
-
import kaolin
|
| 38 |
|
| 39 |
# install nvdiffrast
|
| 40 |
try:
|
| 41 |
import nvdiffrast
|
| 42 |
except ModuleNotFoundError:
|
| 43 |
install_nvdiffrast()
|
| 44 |
-
import nvdiffrast
|
| 45 |
-
|
| 46 |
-
# install diffoctreerast
|
| 47 |
-
try:
|
| 48 |
-
import diffoctreerast
|
| 49 |
-
except ModuleNotFoundError:
|
| 50 |
-
install_diffoctreerast()
|
| 51 |
-
import diffoctreerast
|
| 52 |
|
| 53 |
# install vox2seq
|
| 54 |
try:
|
| 55 |
import vox2seq
|
| 56 |
except ModuleNotFoundError:
|
| 57 |
install_vox2seq()
|
| 58 |
-
import vox2seq
|
| 59 |
|
| 60 |
os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "gradio_temp")
|
| 61 |
os.makedirs(os.environ["GRADIO_TEMP_DIR"], exist_ok=True)
|
|
|
|
| 6 |
|
| 7 |
from scripts.check_packages import (
|
| 8 |
install_flash_attn,
|
|
|
|
| 9 |
install_kaolin,
|
| 10 |
install_nvdiffrast,
|
|
|
|
| 11 |
install_vox2seq
|
| 12 |
)
|
| 13 |
|
|
|
|
| 16 |
import flash_attn
|
| 17 |
except ModuleNotFoundError:
|
| 18 |
install_flash_attn()
|
| 19 |
+
# import flash_attn
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
# install kaolin
|
| 22 |
try:
|
| 23 |
import kaolin
|
| 24 |
except ModuleNotFoundError:
|
| 25 |
install_kaolin()
|
| 26 |
+
# import kaolin
|
| 27 |
|
| 28 |
# install nvdiffrast
|
| 29 |
try:
|
| 30 |
import nvdiffrast
|
| 31 |
except ModuleNotFoundError:
|
| 32 |
install_nvdiffrast()
|
| 33 |
+
# import nvdiffrast
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
# install vox2seq
|
| 36 |
try:
|
| 37 |
import vox2seq
|
| 38 |
except ModuleNotFoundError:
|
| 39 |
install_vox2seq()
|
| 40 |
+
# import vox2seq
|
| 41 |
|
| 42 |
os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), "gradio_temp")
|
| 43 |
os.makedirs(os.environ["GRADIO_TEMP_DIR"], exist_ok=True)
|
scripts/check_packages.py
CHANGED
|
@@ -20,15 +20,6 @@ def install_nvdiffrast():
|
|
| 20 |
result = os.system('pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation')
|
| 21 |
print (result)
|
| 22 |
|
| 23 |
-
def install_diffoctreerast():
|
| 24 |
-
print ("installing diffoctreerast...")
|
| 25 |
-
os.system('mkdir tmp')
|
| 26 |
-
os.system('mkdir tmp/extensions')
|
| 27 |
-
os.system('git clone --recurse-submodules https://github.com/JeffreyXiang/diffoctreerast.git ./tmp/extensions/diffoctreerast')
|
| 28 |
-
result = os.system('pip install ./tmp/extensions/diffoctreerast')
|
| 29 |
-
os.system('rm -rf ./tmp')
|
| 30 |
-
print (result)
|
| 31 |
-
|
| 32 |
def install_mipsplatting():
|
| 33 |
print ("installing mipsplatting...")
|
| 34 |
os.system('mkdir tmp')
|
|
|
|
| 20 |
result = os.system('pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation')
|
| 21 |
print (result)
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
def install_mipsplatting():
|
| 24 |
print ("installing mipsplatting...")
|
| 25 |
os.system('mkdir tmp')
|