DuyTa
/

Vietnamese_ASR

Model card Files Files and versions Metrics Training metrics Community

DuyTa commited on Aug 28, 2023

Commit

5f80484

•

1 Parent(s): a9c93ba

Update realtime_Lora.py

Files changed (1) hide show

realtime_Lora.py +2 -1

realtime_Lora.py CHANGED Viewed

@@ -48,9 +48,10 @@ def main():
     model = PeftModel.from_pretrained(model, peft_model_id)
     model.to("cuda:0")
     pipe = AutomaticSpeechRecognitionPipeline(model=model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, batch_size=8, torch_dtype=torch.float32, device="cuda:0")
-    processor = WhisperProcessor.from_pretrained(peft_config.base_model_name_or_path, language=language, task=task)
     record_timeout = default_record_timeout

     model = PeftModel.from_pretrained(model, peft_model_id)
     model.to("cuda:0")
+    processor = WhisperProcessor.from_pretrained(peft_config.base_model_name_or_path, language=language, task=task)
     pipe = AutomaticSpeechRecognitionPipeline(model=model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, batch_size=8, torch_dtype=torch.float32, device="cuda:0")
     record_timeout = default_record_timeout