Unable to set device type to GPU while using llm-guard[onnxruntime-gpu] for optimization

[manali04shyam]

Hi Team,
I have followed the below documentation for optimization of llm guard to use GPU using onnxruntime-gpu library but even after that during code compilation I see device type as "cpu". Are there any additional steps in the code to enable gpu for llm guard in order to improve speed?

https://llm-guard.com/tutorials/optimization/#onnx-runtime