Spaces:
Build error
Build error
Optimized the app (#1)
Browse files- Optimized the app (0bc51a7f1cce543f88e384377fd565274269506c)
Co-authored-by: Merve Noyan <merve@users.noreply.huggingface.co>
app.py
CHANGED
@@ -2,7 +2,7 @@ from huggingface_hub import from_pretrained_keras
|
|
2 |
import keras_cv
|
3 |
import gradio as gr
|
4 |
|
5 |
-
|
6 |
# load keras model
|
7 |
resolution = 512
|
8 |
dreambooth_model = keras_cv.models.StableDiffusion(
|
|
|
2 |
import keras_cv
|
3 |
import gradio as gr
|
4 |
|
5 |
+
keras.mixed_precision.set_global_policy("mixed_float16")
|
6 |
# load keras model
|
7 |
resolution = 512
|
8 |
dreambooth_model = keras_cv.models.StableDiffusion(
|