Richard
Results
3
comments of
Richard
经过debug后，发现在UNetModel中进行forward时会卡住，具体原因未知： 代码： 类class UNetModel(nn.Layer) 函数 forward for module in self.input_blocks: h = module(h, emb) #此处出现卡死自动退出 hs.append(h)
我是直接调用的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 时会出现上述问题