chansung commited on
Commit
7fa9484
·
1 Parent(s): 46143a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.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)
 
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)