Spaces:

keras-dreambooth
/

voyager

Runtime error

Melanit commited on Mar 25, 2023

Commit

4733b0e

•

1 Parent(s): f5e0820

Update model to newer version

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("melanit/dreambooth_voyager_v2")
 dreambooth_model._diffusion_model = loaded_diffusion_model
 def generate_images(prompt: str, negative_prompt:str, batch_size: int, num_steps: int, guidance_scale: float):

 dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True,
     )
+loaded_diffusion_model = from_pretrained_keras("melanit/dreambooth_voyager_v3")
 dreambooth_model._diffusion_model = loaded_diffusion_model
 def generate_images(prompt: str, negative_prompt:str, batch_size: int, num_steps: int, guidance_scale: float):