transformers
da1af21d - PegasusX add _no_split_modules (#25933)

Commit
2 years ago
PegasusX add _no_split_modules (#25933) * no_split_modules * no_split_modules * inputs_embeds+pos same device * update _no_split_modules * update _no_split_modules
Author
Parents
Loading