kevinwang676 commited on
Commit
d39f6f2
1 Parent(s): 9020641

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -366,7 +366,7 @@ def load_hubert():
366
  global hubert_model
367
 
368
  # Load the model state dictionary from the file
369
- state_dict = torch.load("hubert_base.pt", map_location="cuda:0")
370
 
371
  # Initialize the model
372
  from fairseq.models.hubert import HubertModel
 
366
  global hubert_model
367
 
368
  # Load the model state dictionary from the file
369
+ state_dict = torch.load("hubert_base.pt", map_location="cpu")
370
 
371
  # Initialize the model
372
  from fairseq.models.hubert import HubertModel