UPDATE README.md
Browse files
README.md
CHANGED
@@ -9,6 +9,7 @@ Official huggingface models of **ViTamin**, from the following paper:
|
|
9 |
Load from HuggingFace:
|
10 |
```python
|
11 |
import torch
|
|
|
12 |
from PIL import Image
|
13 |
from transformers import AutoModel, CLIPImageProcessor
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
9 |
Load from HuggingFace:
|
10 |
```python
|
11 |
import torch
|
12 |
+
import open_clip
|
13 |
from PIL import Image
|
14 |
from transformers import AutoModel, CLIPImageProcessor
|
15 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|