Update README.md (#1)
Browse files- Update README.md (6e981474697082083ad1d369c002911cf741b06e)
Co-authored-by: shuai wang <[email protected]>
README.md
CHANGED
|
@@ -30,7 +30,7 @@ For batch processing, the model takes as input
|
|
| 30 |
The model compresses the questions into context embeddings and answers the question based on the provided context embeddings.
|
| 31 |
|
| 32 |
```python
|
| 33 |
-
from transformers import
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained('naver/cocom-v1-16-mistral-7b', trust_remote_code=True)
|
| 36 |
model = model.to('cuda')
|
|
|
|
| 30 |
The model compresses the questions into context embeddings and answers the question based on the provided context embeddings.
|
| 31 |
|
| 32 |
```python
|
| 33 |
+
from transformers import AutoModel
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained('naver/cocom-v1-16-mistral-7b', trust_remote_code=True)
|
| 36 |
model = model.to('cuda')
|