Commit Graph

8 Commits

Author SHA1 Message Date
tsukumi
5ca6e184d7 Fix: ONNX inference of BERT models fails in some Windows environments 2024-09-23 11:33:17 +09:00
tsukumi
b7cd512978 Fix: During ONNX inference, the tensor of the return value of the tokenizer is now obtained in Numpy's NDArray format to eliminate the dependency on PyTorch 2024-09-23 07:40:17 +09:00
tsukumi
dcca37b9a8 Add: Support for speech synthesis in English and Chinese for ONNX inference in Non-JP-Extra models 2024-09-23 07:14:44 +09:00
tsukumi
0e71d7ef7c Fix: Specifying the device_map option in extract_bert_feature() 2024-09-17 05:22:30 +09:00
tsukumi
bdf20e8911 Refactor: Use bert_models.transfer_model() 2024-09-17 05:21:03 +09:00
tsukumi
9c233630ef Refactor: don't keep models in bert_feature module for each language
BERT models and tokenizers are already stored and managed in the bert_models module and should not be stored here.
In addition, since there may be situations where the user would like to use cpu instead of mps for inference when using it as a library, the automatic switching process to mps was removed.
2024-03-10 14:18:57 +00:00
tsukumi
a672aeefd9 Fix: maintain compatibility with Python 3.9 2024-03-08 08:45:56 +00:00
tsukumi
fac4f9a8ab Refactor: rename text_processing to nlp
"text_processing" is clearer, but the import statement is longer.
"nlp" is shorter and makes it clear that it is natural language processing.
2024-03-08 06:20:44 +00:00