PaddleHub 初始化模型是退出

j8ag8udp  于 5个月前  发布在  其他
关注(0)|答案(2)|浏览(42)

欢迎您反馈PaddleHub使用问题,非常感谢您对PaddleHub的贡献!
在留下您的问题时,辛苦您同步提供如下信息:

  • 版本、环境信息

1)PaddleHub 2.3.1 PaddlePaddle版本:2.4.1
2)系统环境:windows 10,python版本 3.7.4 硬件是笔记本,显存4个g

  • 复现信息:
  • module = hub.Module(name='stable_diffusion')

执行到这个代码时退出了,没有错误提示,使用cpu的版本是正常的
里面应该是执行到get_up_block 时闪退的,再具体就没看了

cdmah0mi

cdmah0mi1#

这个模型所需的显存非常大,可能是显存不足的原因。

5us2dqdw

5us2dqdw2#

恩,我也猜是,我看模型大小是4个g多点,但想知 prop 体是不是。然后就是退出没日志输出,感觉不友好,或者是哪个地方还能看这个问题的日志。

相关问题