update the readme to fix the image-to-image example
#18
by
vikasp
- opened
- add
import torch - include
pipe.to("cuda")
Thanks!
patrickvonplaten
changed pull request status to
merged
import torchpipe.to("cuda")Thanks!