Gengzigang
commited on
Commit
·
f489540
1
Parent(s):
cbda8ca
fix
Browse files
README.md
CHANGED
|
@@ -62,6 +62,9 @@ from transformers import AutoModel, AutoConfig, AutoTokenizer
|
|
| 62 |
from transformers import CLIPImageProcessor
|
| 63 |
import torch
|
| 64 |
from llm2vec import LLM2Vec
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
processor = CLIPImageProcessor.from_pretrained("openai/clip-vit-large-patch14-336")
|
| 67 |
model_name_or_path = "microsoft/LLM2CLIP-Openai-L-14-336" # or /path/to/local/LLM2CLIP-Openai-L-14-336
|
|
|
|
| 62 |
from transformers import CLIPImageProcessor
|
| 63 |
import torch
|
| 64 |
from llm2vec import LLM2Vec
|
| 65 |
+
import os
|
| 66 |
+
|
| 67 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
| 68 |
|
| 69 |
processor = CLIPImageProcessor.from_pretrained("openai/clip-vit-large-patch14-336")
|
| 70 |
model_name_or_path = "microsoft/LLM2CLIP-Openai-L-14-336" # or /path/to/local/LLM2CLIP-Openai-L-14-336
|