AUTOMATIC1111 / stable-diffusion-webui-tensorrt

MIT License
311 stars 20 forks source link

found two devices, cpu and cuda:0! #27

Open acergelff opened 1 year ago

acergelff commented 1 year ago

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-h commented 1 year ago

oh yeah, disable all vram optimizations; some sort of issue when converting it

Thelastred01 commented 1 year ago

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