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.