Remove stray cuda call
#1
by
justbruno - opened
A stray cuda call is preventing this model from being run on machines without a GPU.
This has been changed to a dynamically-chosen device allocation, to match input_ids' location.
justbruno changed pull request status to
closed
I realised this could be done more cleanly, and HF does not allow PR edits.