Spaces:

keras-dreambooth
/

marvin_paranoid_android

Runtime error

johko commited on Mar 20, 2023

Commit

91d8eda

1 Parent(s): 48a541f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ resolution = 512
 dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True,
     )
-loaded_diffusion_model = from_pretrained_keras("johko/dreambooth_marvin_paranoid_android")
 dreambooth_model._diffusion_model = loaded_diffusion_model

 dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True,
     )
+loaded_diffusion_model = from_pretrained_keras("keras-dreambooth/marvin_paranoid_android")
 dreambooth_model._diffusion_model = loaded_diffusion_model