there’s an example of snapcraft.yaml with some modifications right here:
Note how KERNEL_IMAGE_FORMAT is set there.