Update README.md

README.md

@@ -76,9 +76,12 @@ European Championship => competitions
 
 If you want to use flash attention or increase sequence length, please, check the following code:
 ```python
+from gliner import GLiNER
+import torch
+
 model = GLiNER.from_pretrained("knowledgator/gliner-qwen-1.5B-v1.0",
                                 _attn_implementation = 'flash_attention_2',
-                                                max_len = 2048).to('cuda:0')
+                                                max_len = 2048).to('cuda:0', dtype=torch.float16)
 ```
 
 ### Benchmarks