Update app.py

app.py

@@ -60,15 +60,15 @@ logger.info("Model downloaded to: %s", model_path)
 
 # Load pipeline
 logger.info('Loading ControlNet model.')
-cache_dir = "./model_cache"  # Customize your cache directory
+cache_dir = "./model_cache"  
 controlnet = FluxControlNetModel.from_pretrained(
-    "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.float16,cache_dir=cache_dir,low_cpu_mem_usage=True
+    "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16,cache_dir=cache_dir
 ).to(device)
 logger.info("ControlNet model loaded successfully.")
 
 logger.info('Loading pipeline.')
 pipe = FluxControlNetPipeline.from_pretrained(
-    model_path, controlnet=controlnet, torch_dtype=torch.float16,cache_dir=cache_dir,low_cpu_mem_usage=True
+    model_path, controlnet=controlnet, torch_dtype=torch.bfloat16,cache_dir=cache_dir
 ).to(device)
 logger.info("Pipeline loaded successfully.")