device_map = 'auto' for parallel model training
#15
by nirku - opened
ValueError: JinaBertModel does not support device_map='auto'. To implement support, the model class needs to implement the _no_split_modules attribute.
Thanks for the bug report!
Should be fixed by https://huggingface.co/jinaai/jina-bert-implementation/discussions/8
nirku changed discussion status to closed