Update app.py

app.py

@@ -12,9 +12,6 @@ from PIL import Image
 import numpy as np
 import gc
 
-# Set default dtype to float16
-jax.config.update("jax_default_dtype", "float16")
-
 # Set up cache directories
 cache_dir = "/tmp/huggingface_cache"
 model_cache_dir = os.path.join(cache_dir, "stable_diffusion_model")