I am new to the platform and also for using GPU. Earlier, I have used the Textattack library for one of my projects using sklean and keras models. For that, I created the customModelWrappers according to the preprocessing and the models, and they worked fine.