Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
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)
|