lmzjms commited on
Commit
f57acc9
1 Parent(s): 9ada147

Update audio_to_text/inference_waveform.py

Browse files
audio_to_text/inference_waveform.py CHANGED
@@ -59,7 +59,7 @@ def decode_caption(word_ids, vocabulary):
59
 
60
 
61
  class AudioCapModel(object):
62
- def __init__(self,weight_dir,device='cuda'):
63
  config = os.path.join(weight_dir,'config.yaml')
64
  self.config = train_util.parse_config_or_kwargs(config)
65
  checkpoint = os.path.join(weight_dir,'swa.pth')
 
59
 
60
 
61
  class AudioCapModel(object):
62
+ def __init__(self,weight_dir,device='cpu'):
63
  config = os.path.join(weight_dir,'config.yaml')
64
  self.config = train_util.parse_config_or_kwargs(config)
65
  checkpoint = os.path.join(weight_dir,'swa.pth')