Ahmet Yildirim
commited on
Commit
·
85e291a
1
Parent(s):
7f9c179
- Update readme
Browse files
README.md
CHANGED
|
@@ -51,7 +51,7 @@ humit_tagger = AutoModel.from_pretrained("Humit-Oslo/humit-tagger-small", trust_
|
|
| 51 |
|
| 52 |
While creating the model, batch\_size and device can be given as parameters:
|
| 53 |
```python
|
| 54 |
-
humit_tagger = AutoModel.from_pretrained("Humit-Oslo/humit-tagger-
|
| 55 |
```
|
| 56 |
|
| 57 |
Here the batch size can be a power of 2, and can be set to higher values, such as 32 or 64, if the model will be loaded on a powerful GPU.
|
|
|
|
| 51 |
|
| 52 |
While creating the model, batch\_size and device can be given as parameters:
|
| 53 |
```python
|
| 54 |
+
humit_tagger = AutoModel.from_pretrained("Humit-Oslo/humit-tagger-small", trust_remote_code=True, batch_size=16, device="cuda")
|
| 55 |
```
|
| 56 |
|
| 57 |
Here the batch size can be a power of 2, and can be set to higher values, such as 32 or 64, if the model will be loaded on a powerful GPU.
|