我正在使用 keras_cv.models.StableDiffusion 从文本生成图像，并具有以下代码：import keras_cvimport kerasimport matplotlib.pyplot as pltfrom PIL import Imagedef plot_images(

我正在使用 keras_cv.models.StableDiffusion 从文本生成图像，并有以下代码：

import keras_cv
import keras
import matplotlib.pyplot as plt
from PIL import Image

def plot_images(images):
    plt.figure(figsize=(20, 20))
    for i in range(len(images)):
        ax = plt.subplot(1, len(images), i + 1)
        plt.imshow(images[i])
        plt.axis("off")

model = keras_cv.models.StableDiffusion(
    img_width=512, img_height=512, jit_compile=False
)

model.save("my_stable_diffusion_model.keras")

images = model.text_to_image("photograph of an astronaut riding a horse", batch_size=3)

Image.fromarray(images[0]).save("astronaut.png")
plot_images(images)

但是，我收到以下错误消息：

model.save("my_stable_diffusion_model.keras")
AttributeError: 'StableDiffusion' object has no attribute 'save'

正在加载的模型非常大，所以我不想每次重新启动程序时都从互联网重新加载它。是否可以将模型及其权重保存在本地，以便我可以重复使用它？

谢谢，

道格

• keras-cv