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
