finetuning_whisper_peft_vi

Runtime error

Namkoy commited on Feb 21

Commit

059c628

•

1 Parent(s): 0975118

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"