carefree0910 / carefree-creator

AI magics meet Infinite draw board.

Home Page:https://creator.nolibox.com/guest

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

iwasnotyet opened this issue · comments

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

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

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

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

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

十分感谢!