@ -35,7 +35,13 @@ from my_utils import load_audio
from tools.i18n.i18n import I18nAuto
i18n = I18nAuto()
device = "cuda"
if torch.cuda.is_available():
elif torch.mps.is_available():
device = "mps"
else:
device = "cpu"
tokenizer = AutoTokenizer.from_pretrained(bert_path)
bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
if is_half == True: