Hugging Face's logo

Spaces:
keras-dreambooth
/
keras-dreambooth-riffusion-currulao
Runtime error

App Files Community
2
juancopi81 commited on
Commit
55a846f
•
1 Parent(s): b819a56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from stable_diffusion_tf.stable_diffusion import StableDiffusion as StableDiffus
2
  import gradio as gr
3
  from tensorflow import keras
4
 
5
- keras.mixed_precision.set_global_policy("mixed_float16")
6
  # load keras model
7
  resolution=512
8
  sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights=False, jit_compile=True)
 
2
  import gradio as gr
3
  from tensorflow import keras
4
 
5
+ keras.mixed_precision.set_global_policy("float32")
6
  # load keras model
7
  resolution=512
8
  sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights=False, jit_compile=True)