finetuning_whisper_peft_vi

Runtime error

Namkoy commited on Feb 21

Commit

0975118

•

1 Parent(s): aa013f1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers import (
 )
 from peft import PeftModel, PeftConfig
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 peft_model_id = "Namkoy/whisper_peft_vi_nam"

 )
 from peft import PeftModel, PeftConfig
+# os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 peft_model_id = "Namkoy/whisper_peft_vi_nam"