Update app.py

app.py

@@ -9,7 +9,8 @@ from diffusers.models.attention_processor import AttnProcessor2_0
 from custom_pipeline import FluxWithCFGPipeline
 
 torch.backends.cuda.matmul.allow_tf32 = True
-
+torch.backends.cudnn.allow_tf32 = True
+torch.set_float32_matmul_precision('high')
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -163,4 +164,4 @@ with gr.Blocks() as demo:
         )
 
 # Launch the app
-demo.launch()
+demo.launch()