You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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)
The text was updated successfully, but these errors were encountered:
The text was updated successfully, but these errors were encountered: