ibm-granite
/

granite-vision-3.3-2b-embedding

Feature Extraction

granitevisionemb

Model card Files Files and versions

Adirazgold commited on Aug 13

Commit

cee615d

·

verified ·

1 Parent(s): e2a0796

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -55,6 +55,7 @@ import requests
 import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModel
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_name = "ibm-granite/granite-vision-3.3-2b-embedding"

 import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModel
+from transformers.utils.import_utils import is_flash_attn_2_available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_name = "ibm-granite/granite-vision-3.3-2b-embedding"