working xformers

app.py

@@ -16,7 +16,6 @@ MAX_IMAGE_SIZE = 2048
 pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16).to("cuda")
 pipe.load_lora_weights("alvdansen/flux-koda")
 pipe.enable_sequential_cpu_offload()
-pipe.enable_xformers_memory_efficient_attention()
 pipe.enable_fp16()
 
 def calculate_optimal_dimensions(image: Image.Image):
@@ -58,6 +57,8 @@ def calculate_optimal_dimensions(image: Image.Image):
 
 @spaces.GPU(durations=300)
 def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+    pipe.enable_xformers_memory_efficient_attention()
+
     image = edit_images["background"]
     width, height = calculate_optimal_dimensions(image)
     mask = edit_images["layers"][0]