Update model_hf.py

model_hf.py

@@ -22,8 +22,9 @@ class SSLModel(nn.Module):
     def __init__(self,device):
         super(SSLModel, self).__init__()
         # eliminate fairseq dependency 
-        repo_id = "facebook/wav2vec2-xlsr-300m"
-        model = Wav2Vec2ForPreTraining.from_pretrained("facebook/wav2vec2-xlsr-300m")
+        # facebook/wav2vec2-xls-r-300m
+        # repo_id = "facebook/wav2vec2-xlsr-300m"
+        model = Wav2Vec2ForPreTraining.from_pretrained("facebook/wav2vec2-xls-r-300m")
         # cp_path = hf_hub_download(repo_id=repo_id, filename=fname)   # Change the pre-trained XLSR model path. 
         # model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path])
         self.model = model[0]