Spaces:

keras-dreambooth
/

seymour-cat-diffusion

Runtime error

AmpleBasis commited on Mar 31, 2023

Commit

d53ef6d

1 Parent(s): f6c70ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ keras.mixed_precision.set_global_policy("mixed_float16")
 # prepare model
 resolution = 512
-sd_dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True,
     )
 db_diffusion_model = from_pretrained_keras("AmpleBasis/seymour-cat")

 # prepare model
 resolution = 512
+sd_dreambooth_model = models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True,
     )
 db_diffusion_model = from_pretrained_keras("AmpleBasis/seymour-cat")