Token Classification
GLiNER
PyTorch
multilingual
NER
GLiNER
information extraction
encoder
entity recognition
Ihor commited on
Commit
20921d4
1 Parent(s): f4c81d5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- max_len = 2048).to('cuda:0', dtype=torch.float16)
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