Update modeling_codegen.py

#4
by Cyrile - opened

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

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