Wuvin commited on
Commit
f37a37f
1 Parent(s): 0c552a7
Files changed (1) hide show
  1. gradio_app/gradio_3dgen.py +5 -0
gradio_app/gradio_3dgen.py CHANGED
@@ -10,8 +10,13 @@ from scripts.refine_lr_to_sr import run_sr_fast
10
  from scripts.utils import save_glb_and_video
11
  from scripts.multiview_inference import geo_reconstruct
12
 
 
 
 
13
  @spaces.GPU
14
  def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"):
 
 
15
  if preview_img is None:
16
  raise gr.Error("preview_img is none")
17
  if isinstance(preview_img, str):
 
10
  from scripts.utils import save_glb_and_video
11
  from scripts.multiview_inference import geo_reconstruct
12
 
13
+
14
+ import nvdiffrast.torch as dr
15
+
16
  @spaces.GPU
17
  def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"):
18
+ dr.RasterizeGLContext(output_db=False) # BUG: cuda_runtime_api.h: No such file or directory
19
+
20
  if preview_img is None:
21
  raise gr.Error("preview_img is none")
22
  if isinstance(preview_img, str):