Spaces:
Build error
Build error
Optimized the app
Browse files
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(
|