Last step doesn't use GPU

[beginjd]

I used the updated last step to test the fine-tuned model. It answers correctly but it doesn't seem to use the GPU.

[MYusufY]

Hi! Just wanted to say that i have opened

• add cuda support to test section and fix pipeline path #3

Which fixes this and some other issues, if you are still interested! :)