test1444 commited on
Commit
2b22791
1 Parent(s): 1d60684

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +6 -0
model.py CHANGED
@@ -5,6 +5,12 @@ import pathlib
5
  import subprocess
6
  import sys
7
 
 
 
 
 
 
 
8
  if os.getenv('SYSTEM') == 'spaces':
9
  import mim
10
 
 
5
  import subprocess
6
  import sys
7
 
8
+ try:
9
+ from mmcv.ops import get_compiling_cuda_version, get_compiler_version
10
+ except:
11
+ import mim
12
+ mim.install('mmcv-full==1.5.0')
13
+
14
  if os.getenv('SYSTEM') == 'spaces':
15
  import mim
16