knowledgator
/

gliner-bi-llama-v1.0

Token Classification

information extraction

entity recognition

Model card Files Files and versions

Ihor commited on Sep 2, 2024

Commit

0e4e201

·

verified ·

1 Parent(s): 1d17cca

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -77,7 +77,7 @@ import torch
 model = GLiNER.from_pretrained("knowledgator/gliner-bi-llama-v1.0",
                                 _attn_implementation = 'flash_attention_2',
-                                                max_len = 2048).to('cuda:0', dtype=torch.float16)
 ```

 model = GLiNER.from_pretrained("knowledgator/gliner-bi-llama-v1.0",
                                 _attn_implementation = 'flash_attention_2',
+                                                max_length = 2048).to('cuda:0', dtype=torch.float16)
 ```