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

Improve: Use the device_map option in transformers to load BERT models directly to the GPU

Browse Source
This commit is contained in:
tsukumi
2024-08-10 16:56:19 +09:00
parent bdf20e8911
commit 80dd6dbc22
7 changed files with 42 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-colab.txt
View File

@@ -1,3 +1,4 @@
accelerate
cmudict
cn2an
g2p_en