一、背景:
执行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)
执行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)