salesforce / LAVIS

LAVIS - A One-stop Library for Language-Vision Intelligence
BSD 3-Clause "New" or "Revised" License
9.67k stars 946 forks source link

Segmentation Fault #660

Open dfan opened 6 months ago

dfan commented 6 months ago

I followed the installation instructions in the README and am running cuda 11.8. However, when I load the model, I get "segmentation fault (core dumped)". This happens whether I specify the device as cpu or cuda.

model, vis_processors, _ = load_model_and_preprocess(name="blip2_t5", model_type="pretrain_flant5xl", is_eval=True, device='cpu')

model, vis_processors, _ = load_model_and_preprocess(name="blip2_t5", model_type="pretrain_flant5xl", is_eval=True, device='cuda:0')

zhao2000hhh commented 2 weeks ago

same question