提示:The size of tensor a (3) must match the size of tensor b (9) at non-singleton dimension 0 #165

一、背景:

执行BLIP blip_model.generate(gpu_image, sample=True, num_beams=3, max_length=20, min_length=5) 生成提示时候产生错误提示如上

二、分析

 这个是包的bug在搜寻 beam search 时候出错

三、解决方法

blip_model.generate(gpu_image, sample=False, num_beams=3, max_length=20, min_length=5)

你可能感兴趣的:(error,fix)