Spaces:
Paused
Paused
Update generate_multitalk.py
Browse files- 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.
|
| 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 |
|