yslan commited on
Commit
815fbc3
1 Parent(s): 8855d72
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import torch
3
  import sys
@@ -43,7 +44,6 @@ os.system('pip install ./submodules/simple-knn')
43
  os.system('pip install ./submodules/diff-surfel-rasterization')
44
 
45
  import argparse
46
- import spaces
47
  import json
48
  import sys
49
  sys.path.append('.')
@@ -424,6 +424,8 @@ if __name__ == "__main__":
424
 
425
  os.environ[
426
  "TORCH_DISTRIBUTED_DEBUG"] = "DETAIL" # set to DETAIL for runtime logging.
 
 
427
 
428
  with open('configs/gradio_i23d_stage2_args.json') as f:
429
  args_2 = json.load(f)
 
1
+ import spaces # RuntimeError: CUDA has been initialized before importing the `spaces` package
2
  import os
3
  import torch
4
  import sys
 
44
  os.system('pip install ./submodules/diff-surfel-rasterization')
45
 
46
  import argparse
 
47
  import json
48
  import sys
49
  sys.path.append('.')
 
424
 
425
  os.environ[
426
  "TORCH_DISTRIBUTED_DEBUG"] = "DETAIL" # set to DETAIL for runtime logging.
427
+ os.environ[
428
+ "XFORMERS_FORCE_DISABLE_TRITON"] = "1" # set to DETAIL for runtime logging.
429
 
430
  with open('configs/gradio_i23d_stage2_args.json') as f:
431
  args_2 = json.load(f)