update
Browse files
README.md
CHANGED
|
@@ -31,6 +31,7 @@ In this paper, we propose LLM2CLIP, a novel approach that embraces the power of
|
|
| 31 |
|
| 32 |
### Pytorch version
|
| 33 |
pip install llm2vec
|
|
|
|
| 34 |
|
| 35 |
Go to [GitHub](https://github.com/microsoft/LLM2CLIP/tree/main/llm2clip)
|
| 36 |
```python
|
|
@@ -41,6 +42,7 @@ from transformers import AutoModel, AutoConfig, AutoTokenizer
|
|
| 41 |
from eva_clip import create_model_and_transforms
|
| 42 |
from llm2vec import LLM2Vec
|
| 43 |
from PIL import Image
|
|
|
|
| 44 |
import torch
|
| 45 |
|
| 46 |
class LLM2VecWrapper(LLM2Vec):
|
|
|
|
| 31 |
|
| 32 |
### Pytorch version
|
| 33 |
pip install llm2vec
|
| 34 |
+
pip install transformers==4.46
|
| 35 |
|
| 36 |
Go to [GitHub](https://github.com/microsoft/LLM2CLIP/tree/main/llm2clip)
|
| 37 |
```python
|
|
|
|
| 42 |
from eva_clip import create_model_and_transforms
|
| 43 |
from llm2vec import LLM2Vec
|
| 44 |
from PIL import Image
|
| 45 |
+
import torch.nn as nn
|
| 46 |
import torch
|
| 47 |
|
| 48 |
class LLM2VecWrapper(LLM2Vec):
|