Spaces:
Running on Zero
Running on Zero
Commit ·
e0a6e7c
1
Parent(s): 2d52c67
online install mipgs
Browse files- app.py +5 -0
- requirements.txt +0 -1
- scripts/check_packages.py +1 -5
app.py
CHANGED
|
@@ -21,6 +21,7 @@ from huggingface_hub import snapshot_download
|
|
| 21 |
from moge.model.v2 import MoGeModel
|
| 22 |
from transformers import AutoModelForMaskGeneration, AutoProcessor
|
| 23 |
from scripts.grounding_sam import plot_segmentation, segment
|
|
|
|
| 24 |
import copy
|
| 25 |
import shutil
|
| 26 |
import time
|
|
@@ -511,6 +512,10 @@ def run_generation(
|
|
| 511 |
|
| 512 |
any_update = False
|
| 513 |
finished_items = []
|
|
|
|
|
|
|
|
|
|
|
|
|
| 514 |
|
| 515 |
for item in pending_exports:
|
| 516 |
if item["future"].done():
|
|
|
|
| 21 |
from moge.model.v2 import MoGeModel
|
| 22 |
from transformers import AutoModelForMaskGeneration, AutoProcessor
|
| 23 |
from scripts.grounding_sam import plot_segmentation, segment
|
| 24 |
+
from scripts.check_packages import install_mipsplatting
|
| 25 |
import copy
|
| 26 |
import shutil
|
| 27 |
import time
|
|
|
|
| 512 |
|
| 513 |
any_update = False
|
| 514 |
finished_items = []
|
| 515 |
+
try:
|
| 516 |
+
import diff_gaussian_rasterization
|
| 517 |
+
except ModuleNotFoundError:
|
| 518 |
+
install_mipsplatting()
|
| 519 |
|
| 520 |
for item in pending_exports:
|
| 521 |
if item["future"].done():
|
requirements.txt
CHANGED
|
@@ -40,4 +40,3 @@ pydantic==2.10.6
|
|
| 40 |
kaolin==0.18.0
|
| 41 |
flash-attn==2.8.3+pt2.8.0cu129
|
| 42 |
nvdiffrast==0.4.0+253ac4fpt2.8.0cu129
|
| 43 |
-
git+https://github.com/ZX-Yin/mip-splatting-submodules.git
|
|
|
|
| 40 |
kaolin==0.18.0
|
| 41 |
flash-attn==2.8.3+pt2.8.0cu129
|
| 42 |
nvdiffrast==0.4.0+253ac4fpt2.8.0cu129
|
|
|
scripts/check_packages.py
CHANGED
|
@@ -26,11 +26,7 @@ def install_nvdiffrast():
|
|
| 26 |
|
| 27 |
def install_mipsplatting():
|
| 28 |
print ("installing mipsplatting...")
|
| 29 |
-
os.system('
|
| 30 |
-
os.system('mkdir tmp/extensions')
|
| 31 |
-
os.system('git clone https://github.com/autonomousvision/mip-splatting.git ./tmp/extensions/mip-splatting')
|
| 32 |
-
result = os.system('pip install ./tmp/extensions/mip-splatting/submodules/diff-gaussian-rasterization/')
|
| 33 |
-
os.system('rm -rf ./tmp')
|
| 34 |
print (result)
|
| 35 |
|
| 36 |
def install_sam2():
|
|
|
|
| 26 |
|
| 27 |
def install_mipsplatting():
|
| 28 |
print ("installing mipsplatting...")
|
| 29 |
+
result = os.system('pip install git+https://github.com/ZX-Yin/mip-splatting-submodules.git --no-build-isolation')
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
print (result)
|
| 31 |
|
| 32 |
def install_sam2():
|