harukaza
harukaza
Thanks for your code!
i also meet the problem, so i add two sentences in the train file. _import torch torch.backends.cudnn.enabled = False_
I solved the problem by using the local opt_model _opt-2.7b_ in the _blip2_pretrain_opt2.7b.yaml_ , which come from Facebook/opt-2.7b.
In fact, I misunderstand the different usages between huggingface and the project page. The former provides a whole ckpt, while the latter need two different module ckpts (i.e., blip2_pretrained_opt2.7b.pth &...