Spaces:
Runtime error
Runtime error
Update utils_app.py
Browse filesJust to make sure it really sets it in right scope 🤞
- utils_app.py +2 -2
utils_app.py
CHANGED
@@ -4,6 +4,8 @@ from tensorflow import keras
|
|
4 |
import tensorflow as tf
|
5 |
import gradio as gr
|
6 |
|
|
|
|
|
7 |
keras_model_list = [
|
8 |
"keras-dreambooth/keras_diffusion_lowpoly_world",
|
9 |
|
@@ -28,8 +30,6 @@ def keras_stable_diffusion(
|
|
28 |
width:int,
|
29 |
):
|
30 |
|
31 |
-
keras.mixed_precision.set_global_policy("mixed_float16")
|
32 |
-
|
33 |
sd_dreambooth_model = models.StableDiffusion(
|
34 |
img_width=height,
|
35 |
img_height=width
|
|
|
4 |
import tensorflow as tf
|
5 |
import gradio as gr
|
6 |
|
7 |
+
keras.mixed_precision.set_global_policy("mixed_float16")
|
8 |
+
|
9 |
keras_model_list = [
|
10 |
"keras-dreambooth/keras_diffusion_lowpoly_world",
|
11 |
|
|
|
30 |
width:int,
|
31 |
):
|
32 |
|
|
|
|
|
33 |
sd_dreambooth_model = models.StableDiffusion(
|
34 |
img_width=height,
|
35 |
img_height=width
|