device error converting SD upscale model on CUDA

[ssube]

Traceback (most recent call last):            
  File "/usr/lib/python3.10/runpy.py", line 196, in _run_module_as_main                                                                                                                               
    return _run_code(code, main_globals, None,                                                                                                                                                        
  File "/usr/lib/python3.10/runpy.py", line 86, in _run_code                                                                                                                                          
    exec(code, run_globals)                                                                                                                                                                           
  File "/home/ssube/onnx-web/api/onnx_web/convert/__main__.py", line 174, in <module>                                                                                                                 
    exit(main())                                                                                                                                                                                      
  File "/home/ssube/onnx-web/api/onnx_web/convert/__main__.py", line 157, in main                                                                                                                     
    load_models(args, ctx, base_models)                                                                                                                                                               
  File "/home/ssube/onnx-web/api/onnx_web/convert/__main__.py", line 94, in load_models                                                                                                               
    convert_diffusion_stable(                                                                                                                                                                         
  File "/home/ssube/onnx-web/api/onnx_env/lib/python3.10/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context                                                                     
    return func(*args, **kwargs)                                                                                                                                                                      
  File "/home/ssube/onnx-web/api/onnx_web/convert/diffusion_stable.py", line 122, in convert_diffusion_stable                                                                                         
    onnx_export(                                                                                                                                                                                      
  File "/home/ssube/onnx-web/api/onnx_web/convert/diffusion_stable.py", line 36, in onnx_export                                                                                                       
    export(                                      
  File "/home/ssube/onnx-web/api/onnx_env/lib/python3.10/site-packages/torch/onnx/utils.py", line 504, in export
    _export(                                     
...
  File "/home/ssube/onnx-web/api/onnx_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1182, in _slow_forward
    result = self.forward(*input, **kwargs)
  File "/home/ssube/onnx-web/api/onnx_env/lib/python3.10/site-packages/torch/nn/modules/sparse.py", line 160, in forward
    return F.embedding(
  File "/home/ssube/onnx-web/api/onnx_env/lib/python3.10/site-packages/torch/nn/functional.py", line 2210, in embedding
    return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument index in method wrapper__index_select)