tsukumi 0e71d7ef7c Fix: Specifying the device_map option in extract_bert_feature()

2024-09-17 05:22:30 +09:00

..

chinese

Fix: Specifying the device_map option in extract_bert_feature()

2024-09-17 05:22:30 +09:00

english

Fix: Specifying the device_map option in extract_bert_feature()

2024-09-17 05:22:30 +09:00

japanese

Fix: Specifying the device_map option in extract_bert_feature()

2024-09-17 05:22:30 +09:00

__init__.py

Format import using isort --profile black --gitignore --lai 2 .

2024-03-11 12:37:51 +09:00

bert_models.py

Improve: Use the device_map option in transformers to load BERT models directly to the GPU

2024-09-17 05:22:06 +09:00

symbols.py

Refactor: rename text_processing to nlp

2024-03-08 06:20:44 +00:00