Update app.py
Browse files
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
|
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
|