PaddleHub disco_diffusion_clip_rn50.reverse_diffusion-2069714227 Output()之后进度条不动

vecaoik1 于 6个月前发布在其他

关注(0)|答案(5)|浏览(59)

paddlepaddle 以及paddlehub版本满足要求
终端：
UserWarning: Setuptools is replacing distutils.
warnings.warn("Setuptools is replacing distutils.")
W0204 12:21:43.541858 19944 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 11.6, Runtime API Version: 11.6
W0204 12:21:43.552976 19944 gpu_resources.cc:91] device: 0, cuDNN Version: 8.4.
disco_diffusion_clip_rn101.reverse_diffusion-828390671
环境安装都没有问题，但是进度条不动并且自动退出
disco_diffusion_clip_rn50.reverse_diffusion - INFO - creating artwork...
disco_diffusion_clip_rn50.reverse_diffusion-2069714227 Output()
0%| | 0/250 [00:00<?, ?it/s]

PaddleHub

来源：https://github.com/PaddlePaddle/PaddleHub/issues/2201

5条答案

按热度按时间

idfiyjo81#

经过debug后，发现在UNetModel中进行forward时会卡住，具体原因未知：
代码：
类class UNetModel(nn.Layer)
函数 forward
for module in self.input_blocks:
h = module(h, emb) #此处出现卡死自动退出
hs.append(h)

赞(0）回复(0）举报 6个月前

93ze6v8z2#

请问你调用的代码或命令是什么？

赞(0）回复(0）举报 6个月前

y4ekin9u3#

我是直接调用的API(和demo里一样)

module = hub.Module(name="disco_diffusion_clip_rn101")
text_prompts = ["A beautiful painting of a singular lighthouse..."]
da = module.generate_image(text_prompts=text_prompts, output_dir='./outcome/',width_height=[512,512])

在运行 module.generate_image 时会出现上述问题

赞(0）回复(0）举报 6个月前

mzaanser4#

一时我也不好判断是不是什么环境问题导致的，这个模型所耗费的显存非常大，一般显存可能带不动。这边建议可以直接在aistudio上用免费显卡进行尝试，我刚刚上了这个项目测了下，发现这个模型在aistudio上还是能跑的。

赞(0）回复(0）举报 6个月前

kdfy810k5#

好的感谢，我去试一下

赞(0）回复(0）举报 6个月前