PyTorch
ssl-aasist
custom_code
ash56 commited on
Commit
cd159a5
·
verified ·
1 Parent(s): 4ab68a5

Update model_hf.py

Browse files
Files changed (1) hide show
  1. model_hf.py +1 -1
model_hf.py CHANGED
@@ -21,7 +21,7 @@ class SSLModel(nn.Module):
21
  def __init__(self,device):
22
  super(SSLModel, self).__init__()
23
 
24
- cp_path = '/export/fs06/agarg22/xlsr_chkpt/xlsr2_300m.pt' # Change the pre-trained XLSR model path.
25
  model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path])
26
  self.model = model[0]
27
  self.model_device=device
 
21
  def __init__(self,device):
22
  super(SSLModel, self).__init__()
23
 
24
+ cp_path = 'xlsr2_300m.pt' # Change the pre-trained XLSR model path.
25
  model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path])
26
  self.model = model[0]
27
  self.model_device=device