Nvidia Container Toolkit in Arion

I’m trying to use Arion to create a docker-compose style environment with GPU acceleration using Nvidia container toolkit in my configuration.nix. How do I tell Arion to enable this option, it seems not the have the necessary options corresponding to the docker-compose option.

            - driver: nvidia
              count: all
              capabilities: [ gpu ]

How do I accomplish this using the configuration.nix (with Arion, or other modules)?
(Additional information. Nvidia container toolkit is installed and running with docker separately, arion is also running with other containers, a root flake is used)