multimodalart HF Staff commited on
Commit
8ee5d5e
·
verified ·
1 Parent(s): d22979d

[Admin maintenance] Migrate to ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,10 +7,13 @@ _omp = os.environ.get("OMP_NUM_THREADS", "")
7
  if _omp and not _omp.isdigit():
8
  os.environ["OMP_NUM_THREADS"] = str(max(1, int("".join(c for c in _omp if c.isdigit()) or "1")))
9
 
 
 
 
 
10
  from nemo.collections.asr.models import ASRModel
11
  import torch
12
  import gradio as gr
13
- import spaces
14
  import gc
15
  import shutil
16
  from pathlib import Path
 
7
  if _omp and not _omp.isdigit():
8
  os.environ["OMP_NUM_THREADS"] = str(max(1, int("".join(c for c in _omp if c.isdigit()) or "1")))
9
 
10
+ # initializing numba with CUDA breaks ZeroGPU
11
+ os.environ.setdefault("NUMBA_DISABLE_CUDA", "1")
12
+
13
+ import spaces
14
  from nemo.collections.asr.models import ASRModel
15
  import torch
16
  import gradio as gr
 
17
  import gc
18
  import shutil
19
  from pathlib import Path