test_input and model can be on different devices

[kaczmarj]

this should be fixed to put the inputs on the same device as the model.

https://github.com/kaczmarj/wsinfer/blob/11580308546e1f75147a276115bde7773e0b9423/wsinfer/_modellib/run_inference.py#L197