Update app.py

app.py

@@ -13,6 +13,14 @@ from diffusers.pipelines.flux.pipeline_flux import FluxPipeline
 from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
+from accelerate import Accelerator
+
+
+# Instantiate the Accelerator
+accelerator = Accelerator()
+
+# Use the accelerator to prepare your models, optimizers, and data loaders
+model, optimizer, dataloader = accelerator.prepare(model, optimizer, dataloader)
 
 dtype = torch.bfloat16