Update README.md
Browse files
README.md
CHANGED
|
@@ -27,7 +27,9 @@ With a compressed collection of documents to retrieve from, inference becomes ab
|
|
| 27 |
## Usage
|
| 28 |
|
| 29 |
```python
|
| 30 |
-
|
|
|
|
|
|
|
| 31 |
|
| 32 |
# Example documents and question:
|
| 33 |
documents = [
|
|
|
|
| 27 |
## Usage
|
| 28 |
|
| 29 |
```python
|
| 30 |
+
from transformers import AutoModel
|
| 31 |
+
|
| 32 |
+
pisco = AutoModel.from_pretrained('naver/pisco-mistral', device_map='cuda')
|
| 33 |
|
| 34 |
# Example documents and question:
|
| 35 |
documents = [
|