Update src/flux/xflux_pipeline.py

src/flux/xflux_pipeline.py

@@ -162,7 +162,7 @@ class XFluxPipeline:
         image_proj = self.improj(image_prompt_embeds)
         return image_proj
 
-    @spaces.GPU
+    @spaces.GPU()
     def __call__(self,
                  prompt: str,
                  image_prompt: Image = None,
@@ -219,7 +219,6 @@ class XFluxPipeline:
             neg_ip_scale=neg_ip_scale,
         )
 
-    @torch.inference_mode()
     def gradio_generate(self, prompt, image_prompt, controlnet_image, width, height, guidance,
                         num_steps, seed, true_gs, ip_scale, neg_ip_scale, neg_prompt,
                         neg_image_prompt, timestep_to_start_cfg, control_type, control_weight,