merve HF staff commited on
Commit
93e53d1
1 Parent(s): b5aaeaf

Update utils_app.py

Browse files

Just to make sure it really sets it in right scope 🤞

Files changed (1) hide show
  1. 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