GLiNER
GLiNER copied to clipboard
Does this model benefit from a GPU
I see no documentation on how to leverage one. Is this because it would not help?
It does! You can see here how to set it onto the GPU for inference: https://github.com/urchade/GLiNER/blob/9068a2f5ae854eb6288c24ae3d7f36e9d84c06ba/gliner/model.py#L107
The device can be set as a model property.
Yes, Gliner is much more performant on GPU. Base on the test done on my side, it will run about 250 times faster on GPU then on CPU. For CPU base, spacy would be a better choice.