HfSpeechUtils
/

convert_wav2vec2_to_hf

Model card Files Files and versions

patrickvonplaten commited on Dec 20, 2021

Commit

14441d6

·

1 Parent(s): 9515498

correct

Files changed (1) hide show

run_forward.py +2 -2

run_forward.py CHANGED Viewed

@@ -10,9 +10,9 @@ import sys
 from shutil import copyfile
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2Model, Wav2Vec2CTCTokenizer, Wav2Vec2FeatureExtractor
-finetuned = bool(int(sys.argv[1]))
 fairseq_wav2vec2_path = str(sys.argv[2])
-hf_path = str(sys.argv[3])
 if finetuned:

 from shutil import copyfile
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2Model, Wav2Vec2CTCTokenizer, Wav2Vec2FeatureExtractor
+hf_path = str(sys.argv[1])
 fairseq_wav2vec2_path = str(sys.argv[2])
+finetuned = bool(int(sys.argv[3]))
 if finetuned: