Kernel crashes while loading pre-trained BLIP-2 model #14108
Labels
bug
Issue identified by VS Code Team member as probable bug
info-needed
Issue requires more information from poster
user-setup
Issues caused by incorrectly configured python environments, packages, or the like
Applies To
What happened?
I tried to run the BLIP-2 model for Image Captioning from this blog https://huggingface.co/blog/blip-2. All cells above it work fine, but as soon as this cell runs for about a minute kernel dies.
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", torch_dtype=torch.float16)
VS Code Version
1.81.1
Jupyter Extension Version
v2023.7.1002162226
Jupyter logs
Coding Language and Runtime Version
Python v3.10.12
Running Jupyter locally or remotely?
Local
The text was updated successfully, but these errors were encountered: