DragGAN icon indicating copy to clipboard operation
DragGAN copied to clipboard
verified publisher icon OpenGVLab

use cpu to inference

Open Fischer-pixel opened this issue 2 years ago • 1 comments

when I change the "device" in gradio_app.py from cuda to cpu,it raises error "expect all tensors on the same device".

Fischer-pixel avatar May 23 '23 10:05 Fischer-pixel

I have made a patch to make it work on cpu.

See f70ab05da2f2765cfe093283761b2919ae5c4b51

However, running on CPU would be very slow.

Zeqiang-Lai avatar May 24 '23 01:05 Zeqiang-Lai