From eee6e422907c23e0d63f49f9e1eaf2e2a615fca8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Tue, 5 Sep 2023 20:24:23 +0800
Subject: [PATCH] Update japanese_bert.py

---
 text/japanese_bert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/text/japanese_bert.py b/text/japanese_bert.py
index 4f349df..d5bfa26 100644
--- a/text/japanese_bert.py
+++ b/text/japanese_bert.py
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained("./bert/bert-base-japanese-v3")
 
-def get_bert_feature(text, word2ph):
+def get_bert_feature(text, word2ph, device=None):
     if sys.platform == "darwin" and torch.backends.mps.is_available() and device == "cpu":
         device = "mps"
     if not device: