kevinwang676 commited on
Commit
743980c
·
verified ·
1 Parent(s): 25c6499

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -358,7 +358,7 @@ def load_hubert():
358
  suffix="",
359
  )
360
  hubert_model = models[0]
361
- hubert_model = hubert_model.to(config.device)
362
  if config.is_half:
363
  hubert_model = hubert_model.half()
364
  else:
 
358
  suffix="",
359
  )
360
  hubert_model = models[0]
361
+ hubert_model = hubert_model.to("cpu")#(config.device)
362
  if config.is_half:
363
  hubert_model = hubert_model.half()
364
  else: