Spaces:
Build error
Build error
Optimized the app
#1
by
merve
HF staff
- opened
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(
|