Hugging Face's logo

Spaces:
cbensimon
/
PASD
Paused

App Files Community
cbensimon HF staff commited on
Commit
fe37283
·
1 Parent(s): 563fa56

Patch torch.cuda.get_device_capability

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import sys; sys.path.append('./PASD')
2
  import spaces
3
  import os
 
1
+ import torch
2
+ torch.cuda.get_device_capability = lambda *args, **kwargs: (8, 6)
3
+
4
  import sys; sys.path.append('./PASD')
5
  import spaces
6
  import os