martylabs commited on
Commit
bfe73fb
·
verified ·
1 Parent(s): d038d91

Update generate_multitalk.py

Browse files
Files changed (1) hide show
  1. generate_multitalk.py +1 -1
generate_multitalk.py CHANGED
@@ -216,7 +216,7 @@ def _parse_args():
216
 
217
  def custom_init(device, wav2vec):
218
  audio_encoder = Wav2Vec2ForCTC.from_pretrained(args.wav2vec_dir, attn_implementation="eager").to(device)
219
- audio_encoder.feature_extractor._freeze_parameters()
220
  wav2vec_feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(wav2vec, local_files_only=True)
221
  return wav2vec_feature_extractor, audio_encoder
222
 
 
216
 
217
  def custom_init(device, wav2vec):
218
  audio_encoder = Wav2Vec2ForCTC.from_pretrained(args.wav2vec_dir, attn_implementation="eager").to(device)
219
+ audio_encoder.freeze_feature_extractor()
220
  wav2vec_feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(wav2vec, local_files_only=True)
221
  return wav2vec_feature_extractor, audio_encoder
222