Update README.md
Browse files
README.md
CHANGED
|
@@ -55,6 +55,7 @@ import requests
|
|
| 55 |
import torch
|
| 56 |
from PIL import Image
|
| 57 |
from transformers import AutoProcessor, AutoModel
|
|
|
|
| 58 |
|
| 59 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 60 |
model_name = "ibm-granite/granite-vision-3.3-2b-embedding"
|
|
|
|
| 55 |
import torch
|
| 56 |
from PIL import Image
|
| 57 |
from transformers import AutoProcessor, AutoModel
|
| 58 |
+
from transformers.utils.import_utils import is_flash_attn_2_available
|
| 59 |
|
| 60 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 61 |
model_name = "ibm-granite/granite-vision-3.3-2b-embedding"
|