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

14 KiB

Raw Blame History

View Raw