我正在使用 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'
正在加载的模型非常大,所以我不想每次重新启动程序时都从互联网重新加载它。是否可以将模型及其权重保存在本地,以便我可以重复使用它?
谢谢,
道格