JasonYinnnn commited on
Commit
e0a6e7c
·
1 Parent(s): 2d52c67

online install mipgs

Browse files
Files changed (3) hide show
  1. app.py +5 -0
  2. requirements.txt +0 -1
  3. 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('mkdir tmp')
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():