merve HF Staff commited on
Commit
3982ff3
·
1 Parent(s): 7517d11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ keras.mixed_precision.set_global_policy("mixed_float16")
9
  # prepare model
10
  resolution = 512
11
  sd_dreambooth_model = models.StableDiffusion(
12
- img_width=resolution, img_height=resolution, jit_compile=True,
13
  )
14
  db_diffusion_model = from_pretrained_keras("merve/dreambooth_diffusion_model")
15
  sd_dreambooth_model._diffusion_model = db_diffusion_model
 
9
  # prepare model
10
  resolution = 512
11
  sd_dreambooth_model = models.StableDiffusion(
12
+ img_width=resolution, img_height=resolution
13
  )
14
  db_diffusion_model = from_pretrained_keras("merve/dreambooth_diffusion_model")
15
  sd_dreambooth_model._diffusion_model = db_diffusion_model