carefree0910 / carefree-creator

AI magics meet Infinite draw board.
https://creator.nolibox.com/guest
MIT License
2.13k stars 216 forks source link

本地部署图片生成正常,但是图像处理(超分辨率)报错:ValueError: Expected a cuda device, but got: cpu #15

Open iwasnotyet opened 1 year ago

iwasnotyet commented 1 year ago

1667678232374

carefree0910 commented 1 year ago

确实是个问题,原因有两个:

  1. empty_cuda_cache 时应该先看看当前 device 是不是 cpu
  2. TranslatorAPI 也需要实现一个 to 方法,从而在需要时可以把它放到 cuda 上执行

我会尽快进行修复,修好后再让你试试😉

carefree0910 commented 1 year ago

刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新 pip install -e . 一下(因为需要更新背后 carefree-learn 的版本);如果还有问题的话,欢迎继续留言😆

iwasnotyet commented 1 year ago

刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新 pip install -e . 一下(因为需要更新背后 carefree-learn 的版本);如果还有问题的话,欢迎继续留言😆

十分感谢!