bbexx commited on
Commit
e4f2f85
1 Parent(s): e7cb1dc

UPDATE README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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"