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
|