From 3719c5e303724d54bd0ef8af14e7d55420c1ddff 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 19:38:45 +0800
Subject: [PATCH] Update chinese_bert.py

---
 text/chinese_bert.py | 10 ----------
 1 file changed, 10 deletions(-)

diff --git a/text/chinese_bert.py b/text/chinese_bert.py
index 38ef7d8..178e36d 100644
--- a/text/chinese_bert.py
+++ b/text/chinese_bert.py
@@ -2,16 +2,6 @@ import torch
 import sys
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
-device = torch.device(
-        "cuda"
-        if torch.cuda.is_available()
-        else (
-            "mps"
-            if sys.platform == "darwin" and torch.backends.mps.is_available()
-            else "cpu"
-        )
-    )
-
 tokenizer = AutoTokenizer.from_pretrained("./bert/chinese-roberta-wwm-ext-large")
 
 def get_bert_feature(text, word2ph, device):