unimol_tools do not make use of gpu when infer

[PoloWitty]

The code implemented in unimol_tools did not pass the cuda variable to Trainer, so the gpu won't be used.

The related code is here.

Should be changed like this:

self.trainer = Trainer(task='repr', cuda=self.device)

[emotionor]

Thank you for bringing this issue to our attention and for providing a solution.

Your contribution is greatly appreciated.

We will review the fix and incorporate it if appropriate. Thanks again for your help!