wlyu-adobe commited on
Commit
46cadc4
·
1 Parent(s): 20f1ef9

Initial commit

Browse files
Files changed (2) hide show
  1. app.py +13 -0
  2. requirements.txt +2 -3
app.py CHANGED
@@ -24,6 +24,19 @@ from einops import rearrange
24
  from PIL import Image
25
  from huggingface_hub import snapshot_download
26
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  from gslrm.model.gaussians_renderer import render_turntable, imageseq2video
28
  from mvdiffusion.pipelines.pipeline_mvdiffusion_unclip import StableUnCLIPImg2ImgPipeline
29
  from utils_folder.face_utils import preprocess_image, preprocess_image_without_cropping
 
24
  from PIL import Image
25
  from huggingface_hub import snapshot_download
26
 
27
+ # Install diff-gaussian-rasterization at runtime (requires GPU)
28
+ import subprocess
29
+ import sys
30
+ try:
31
+ import diff_gaussian_rasterization
32
+ except ImportError:
33
+ print("Installing diff-gaussian-rasterization...")
34
+ subprocess.check_call([
35
+ sys.executable, "-m", "pip", "install",
36
+ "git+https://github.com/graphdeco-inria/diff-gaussian-rasterization"
37
+ ])
38
+ import diff_gaussian_rasterization
39
+
40
  from gslrm.model.gaussians_renderer import render_turntable, imageseq2video
41
  from mvdiffusion.pipelines.pipeline_mvdiffusion_unclip import StableUnCLIPImg2ImgPipeline
42
  from utils_folder.face_utils import preprocess_image, preprocess_image_without_cropping
requirements.txt CHANGED
@@ -8,13 +8,12 @@ accelerate>=0.20.0
8
  opencv-python>=4.8.0
9
  rembg>=2.0.50
10
  facenet-pytorch>=2.5.3
11
- git+https://github.com/graphdeco-inria/diff-gaussian-rasterization
12
  einops>=0.7.0
 
13
  plyfile>=0.9
14
  easydict>=1.10
15
  pyyaml>=6.0
16
  lpips>=0.1.4
17
  huggingface-hub>=0.19.0
18
- videoio>=0.2.0
19
  gradio>=5.0.0
20
- xformers>=0.0.20
 
8
  opencv-python>=4.8.0
9
  rembg>=2.0.50
10
  facenet-pytorch>=2.5.3
 
11
  einops>=0.7.0
12
+ ninja
13
  plyfile>=0.9
14
  easydict>=1.10
15
  pyyaml>=6.0
16
  lpips>=0.1.4
17
  huggingface-hub>=0.19.0
18
+ videoio==0.3.0
19
  gradio>=5.0.0