Move input_ids to cuda device

#9
by freddyaboulton HF staff - opened

The original code does not work on ZeroGPU

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment