Update modeling_codegen.py

#5

Hello, this small change ensures that the labels are on the correct device.

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

Sign up or log in to comment