Update README.md
Browse filesAdd torch package for using transformers
README.md
CHANGED
|
@@ -54,6 +54,7 @@ Granite Vision model is supported natively `transformers` from the `main` branch
|
|
| 54 |
|
| 55 |
```python
|
| 56 |
from transformers import AutoProcessor, AutoModelForVision2Seq
|
|
|
|
| 57 |
|
| 58 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 59 |
|
|
|
|
| 54 |
|
| 55 |
```python
|
| 56 |
from transformers import AutoProcessor, AutoModelForVision2Seq
|
| 57 |
+
import torch
|
| 58 |
|
| 59 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 60 |
|