Move input_ids to cuda device

#9
by freddyaboulton HF staff - opened

The original code does not work on ZeroGPU

Works for me.

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment