knowledgator
/

gliner-llama-1B-v1.0

Token Classification

information extraction

entity recognition

Model card Files Files and versions

Ihor commited on Sep 2, 2024

Commit

34a0b4b

·

verified ·

1 Parent(s): d05dd44

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

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

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