Update README.md
Browse files
README.md
CHANGED
@@ -81,7 +81,7 @@ import torch
|
|
81 |
|
82 |
model = GLiNER.from_pretrained("knowledgator/gliner-qwen-1.5B-v1.0",
|
83 |
_attn_implementation = 'flash_attention_2',
|
84 |
-
|
85 |
```
|
86 |
|
87 |
### Benchmarks
|
|
|
81 |
|
82 |
model = GLiNER.from_pretrained("knowledgator/gliner-qwen-1.5B-v1.0",
|
83 |
_attn_implementation = 'flash_attention_2',
|
84 |
+
max_length = 2048).to('cuda:0', dtype=torch.float16)
|
85 |
```
|
86 |
|
87 |
### Benchmarks
|