found two devices, cpu and cuda:0!
acergelff opened this issue · comments
acergelff commented
When I click on the "Convert Unet to ONNX" button in the "Convert to ONNX" tab, I am prompted with the following error
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_CUDA_addmm
How can I solve this problem?
yoinked commented
oh yeah, disable all vram optimizations; some sort of issue when converting it
Thelastred01 commented
oh yeah, disable all vram optimizations; some sort of issue when converting it
Thanks, I just deleted --medvram from the webui User.bat file and now TensorRT extension works good . Took much time to find this solution