Hugging Face's logo

Spaces:
vibs08
/
image-3d-fastapi
Runtime error

App Files Community
vibs08 commited on
Commit
50b44f7
·
verified ·
1 Parent(s): 5549457

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -43,7 +43,13 @@ HEADER = """FRAME AI"""
43
 
44
  # torch.cuda.set_device(1)
45
 
46
- CUDA_LAUNCH_BLOCKING=1
 
 
 
 
 
 
47
 
48
  model = TSR.from_pretrained(
49
  "stabilityai/TripoSR",
 
43
 
44
  # torch.cuda.set_device(1)
45
 
46
+ # CUDA_LAUNCH_BLOCKING=1
47
+
48
+
49
+ if torch.cuda.is_available():
50
+ torch.cuda.set_device(1)
51
+
52
+
53
 
54
  model = TSR.from_pretrained(
55
  "stabilityai/TripoSR",