Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,8 +20,8 @@ MODEL = from_pretrained_keras(MODEL_CKPT)
|
|
20 |
model = keras_cv.models.StableDiffusion(
|
21 |
img_width=img_width, img_height=img_height, jit_compile=True
|
22 |
)
|
23 |
-
model.
|
24 |
-
model.
|
25 |
|
26 |
# Warm-up the model.
|
27 |
_ = model.text_to_image("Teddy bear", batch_size=num_images_to_gen)
|
|
|
20 |
model = keras_cv.models.StableDiffusion(
|
21 |
img_width=img_width, img_height=img_height, jit_compile=True
|
22 |
)
|
23 |
+
model._text_encoder = MODEL
|
24 |
+
model._text_encoder.compile(jit_compile=True)
|
25 |
|
26 |
# Warm-up the model.
|
27 |
_ = model.text_to_image("Teddy bear", batch_size=num_images_to_gen)
|