Hello everyone, I'm currently working on implementing BentoML's new Controlnet use case and aiming to store all models in our local Bentoml model repository. I'm seeking an example for saving and retrieving the model named 'madebyollin/sdxl-vae-fp16-fix'. Here's the code I have so far: controlnet = ControlNetModel.from_pretrained("diffusers/controlnet-canny-sdxl-1.0") pipe = StableDiffusionXLControlNetPipeline.from_pretrained( "stabilityai/stable-diffusion-xl-base-1.0", controlnet=controlnet ) bentoml.xyz.save_model('controlnet-canny-sdxl-1.0', pipe) Could someone please advise which package we need to utilize for this model so that we can use it? Specifically, I'm interested in how to access it using: vae = bentoml.models.get("sdxl-vae-fp16-fix"). Bentoml 1.2.2 version Thank you in advance!