Ahmet Yildirim commited on
Commit
85e291a
·
1 Parent(s): 7f9c179

- Update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-xs", 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.
 
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.