search

salesforce / LAVIS

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

Segmentation Fault #660

Open dfan opened 9 months ago

dfan commented 9 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 3 months ago

same question

jkim50104 commented 1 day ago

I get this when I use cv2.imshow() ?