How to train SETFIT on a dataset like NLI

Hey there,
I have tried training the SET_FIT model on datasets of the form {text,label}.
The library provides inbuilt function to do the same,
I have a doubt though,
how would we train it on a dataset like the NLI dataset.
Basically a dataset of the form (sent1,sent2,label) where label represents the relation between the two sentences.
Thank you for reading.