Gengzigang commited on
Commit ·
9e2e7d2
1
Parent(s): 767f67f
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/openai/clip-vit-base-patch16")
|
| 67 |
model_name_or_path = "microsoft/LLM2CLIP-Openai-B-16" # or /path/to/local/LLM2CLIP-Openai-B-16
|
|
|
|
| 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/openai/clip-vit-base-patch16")
|
| 70 |
model_name_or_path = "microsoft/LLM2CLIP-Openai-B-16" # or /path/to/local/LLM2CLIP-Openai-B-16
|