I think it is universal across the hugging face API. You can use the fine-tuned trainer to generate predictions. Something like trainer.predict() should work as expected. I haven’t touched the code for a while, let me know if it’s not working for you.