Commit Graph

14 Commits

Author SHA1 Message Date
tsukumi
c1fce3fec7 Improve: Make it possible to convert BERT language models to FP16 2024-12-19 07:16:59 +09:00
tsukumi
4a92a1abad Fix: Missing onnxsim dependency 2024-11-13 10:50:44 +09:00
tsukumi
f135a79316 Fix: Missing onnx dependency 2024-11-13 10:43:58 +09:00
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
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
a9c5d18eec Try to fix colab torch version 2024-09-09 16:30:44 +09:00
litagin02
38adf43924 Fix: colab torchvision bug, explicit torchaudio version 2024-09-09 15:31:30 +09:00
litagin02
7a99e3e1b7 Update requirements-colab.txt 2024-07-27 12:32:13 +09:00
litagin02
52bb5dc56d Fix: require numpy<2 for numpy major version up 2024-06-17 06:54:50 +09:00
litagin02
5dd7db10ce Require gradio>=4.32 due to render decorator 2024-06-16 18:45:47 +09:00
litagin02
287db8cf28 Fix: add onnxruntime to colab req 2024-06-01 15:18:50 +09:00
litagin02
e66074c243 Fix: add torch to req 2024-05-29 07:06:00 +09:00
litagin02
ceb431192c colab 2024-05-28 18:37:44 +09:00