whaohan commited on
Commit
c70956c
·
verified ·
1 Parent(s): 09d6c31
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ from huggingface_hub import snapshot_download
16
  from model.model import MeshTransformer
17
  from utils import apply_normalize, joint_filter, sample_pc
18
 
 
19
 
20
  CONFIG_PATH = 'config/BPT-open-8k-8-16.yaml'
21
  with open(CONFIG_PATH, "r") as f:
 
16
  from model.model import MeshTransformer
17
  from utils import apply_normalize, joint_filter, sample_pc
18
 
19
+ os.environ["CUDA_HOME"] = "/usr/local/cuda"
20
 
21
  CONFIG_PATH = 'config/BPT-open-8k-8-16.yaml'
22
  with open(CONFIG_PATH, "r") as f: