Commit Graph

8 Commits

Author SHA1 Message Date
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