tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix: I'm a Python beginner

Browse Source
This commit is contained in:
litagin02
2023-12-30 20:07:43 +09:00
parent 9045fa3426
commit 4faef57dd2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
text/english_bert_mock.py
View File

@@ -28,7 +28,7 @@ def get_bert_feature(
device = "mps" device = "mps"
if not device: if not device:
device = "cuda" device = "cuda"
if device = "cuda" and not torch.cuda.is_available(): if device == "cuda" and not torch.cuda.is_available():
device = "cpu" device = "cpu"
if device not in models.keys(): if device not in models.keys():
models[device] = DebertaV2Model.from_pretrained(LOCAL_PATH).to(device) models[device] = DebertaV2Model.from_pretrained(LOCAL_PATH).to(device)