Tuananh20015 commited on
Commit
08d3032
·
verified ·
1 Parent(s): 64deef9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,10 +25,10 @@ from omnivoice.cli.demo import build_demo
25
  # ---------------------------------------------------------------------------
26
  CHECKPOINT = os.environ.get("OMNIVOICE_MODEL", "k2-fsa/OmniVoice")
27
 
28
- print(f"Loading model from {CHECKPOINT} to cuda ...")
29
  model = OmniVoice.from_pretrained(
30
  CHECKPOINT,
31
- device_map="cuda",
32
  dtype=torch.float16,
33
  load_asr=True,
34
  )
 
25
  # ---------------------------------------------------------------------------
26
  CHECKPOINT = os.environ.get("OMNIVOICE_MODEL", "k2-fsa/OmniVoice")
27
 
28
+ print(f"Loading model from {CHECKPOINT} to cpu ...")
29
  model = OmniVoice.from_pretrained(
30
  CHECKPOINT,
31
+ device_map="cpu",
32
  dtype=torch.float16,
33
  load_asr=True,
34
  )