LeonardoImageDriver isn't outputting proper images anymore

[shhlife]

Users have been saying that images generated by Leonardo aren't working in comfyUI, so I tested with the framework and I'm getting images out that aren't displayable.

import os

from griptape.drivers import LocalFileManagerDriver
from griptape.drivers.image_generation.leonardo import LeonardoImageGenerationDriver
from griptape.structures import Agent
from griptape.tools import FileManagerTool, PromptImageGenerationTool

driver = LeonardoImageGenerationDriver(
    model="b24e16ff-06e3-43eb-8d33-4416c2d75876",
    api_key=os.environ["LEONARDO_API_KEY"],
    image_width=512,
    image_height=1024,
)


agent = Agent(
    tools=[
        PromptImageGenerationTool(image_generation_driver=driver),
        FileManagerTool(file_manager_driver=LocalFileManagerDriver(workdir="images")),
    ],
)

agent.run(
    "Generate a watercolor painting of a dog riding a skateboard and save it as dog.png"
)

The image that came out couldn't be displayed.