Leveraging GPU to run the quickstart_vision.ipynb notebook

[Analect]

Hi. Thanks for your work on this.
I was trying to run your example https://github.com/Giskard-AI/giskard/blob/main/docs/getting_started/quickstart/quickstart_vision.ipynb on Colab. Initially I was using the default CPU runtime, but the notebook ran for >1hr and eventually timed-out. I was then trying to run it on a Colab GPU runtime (T4). It says "To use the fast image processor class set use_fast=True." Could you advise how the model setup in the sample notebook should be adjusted to be able to leverage a GPU?

Fast image processor class <class 'transformers.models.vit.image_processing_vit_fast.ViTImageProcessorFast'> is available for this model. Using slow image processor class. To use the fast image processor class set `use_fast=True`.
Device set to use cpu