gensym commited on
Commit
eee56a3
1 Parent(s): 98ade97
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -19,6 +19,11 @@ print("importing easydict")
19
  from easydict import EasyDict as edict
20
  print("importing PIL")
21
  from PIL import Image
 
 
 
 
 
22
  print("importing trellis image to 3d pipeline")
23
  from trellis.pipelines import TrellisImageTo3DPipeline
24
  print("importing trellis representations")
@@ -27,10 +32,6 @@ print("importing trellis utils")
27
  from trellis.utils import render_utils, postprocessing_utils
28
 
29
 
30
- print(f"Is CUDA available: {torch.cuda.is_available()}")
31
- # True
32
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
33
-
34
  MAX_SEED = np.iinfo(np.int32).max
35
  TMP_DIR = "/tmp/Trellis-demo"
36
 
 
19
  from easydict import EasyDict as edict
20
  print("importing PIL")
21
  from PIL import Image
22
+
23
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
24
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
25
+
26
+
27
  print("importing trellis image to 3d pipeline")
28
  from trellis.pipelines import TrellisImageTo3DPipeline
29
  print("importing trellis representations")
 
32
  from trellis.utils import render_utils, postprocessing_utils
33
 
34
 
 
 
 
 
35
  MAX_SEED = np.iinfo(np.int32).max
36
  TMP_DIR = "/tmp/Trellis-demo"
37