Commit Graph

10 Commits

Author SHA1 Message Date
tsukumi
fe253611d3 Fix: nltk to 3.8.1 or lower
Due to reports that updating nltk to 3.8.2 or later significantly slows down learning, including English.
2024-09-23 06:22:47 +09:00
tsukumi
b95420325b Improve: Changed PyTorch to an optional dependency when used style_bert_vits2 as a library
ONNXRuntime has relied on CUDA 12.x / cuDNN 9.x by default since v1.19.0, while PyTorch finally supports cuDNN 9 since v2.4.0.
Therefore, unless the pinning of torch<2.4 is eliminated, CUDA cannot be used with the latest version of ONNXRuntime.
2024-09-17 17:56:59 +09:00
tsukumi
eede796c4e Add: Preparation for ONNX inference support ① 2024-09-17 06:53:11 +09:00
tsukumi
80dd6dbc22 Improve: Use the device_map option in transformers to load BERT models directly to the GPU 2024-09-17 05:22:06 +09:00
litagin02
e8367cc4e2 Update requirements-infer.txt 2024-07-27 12:31:53 +09:00
litagin02
a1fcb433e7 Fix numpy<2 2024-06-17 21:49:00 +09:00
litagin02
52bb5dc56d Fix: require numpy<2 for numpy major version up 2024-06-17 06:54:50 +09:00
litagin02
f80846b223 Fix infer req and not to use virtualenv 2024-06-01 14:08:56 +09:00
litagin02
e66074c243 Fix: add torch to req 2024-05-29 07:06:00 +09:00
litagin02
9625014184 Feat: use uv for bat installer 2024-05-29 06:53:21 +09:00