Update app.py
Browse files
app.py
CHANGED
|
@@ -22,8 +22,8 @@ weights_path = keras.utils.get_file(
|
|
| 22 |
pokemon_model = keras_cv.models.StableDiffusion(
|
| 23 |
img_width=img_width, img_height=img_height
|
| 24 |
)
|
| 25 |
-
pokemon_model.diffusion_model.load_weights(weights_path)
|
| 26 |
pokemon_model.to(device)
|
|
|
|
| 27 |
|
| 28 |
pokemon_model.diffusion_model.compile(jit_compile=True)
|
| 29 |
pokemon_model.decoder.compile(jit_compile=True)
|
|
|
|
| 22 |
pokemon_model = keras_cv.models.StableDiffusion(
|
| 23 |
img_width=img_width, img_height=img_height
|
| 24 |
)
|
|
|
|
| 25 |
pokemon_model.to(device)
|
| 26 |
+
pokemon_model.diffusion_model.load_weights(weights_path)
|
| 27 |
|
| 28 |
pokemon_model.diffusion_model.compile(jit_compile=True)
|
| 29 |
pokemon_model.decoder.compile(jit_compile=True)
|