pipeline update

pipeline.py

@@ -6939,7 +6939,6 @@ class InterleaveInferencer:
                 output_dict['text'] = i
         return output_dict
         
-from diffusers import DiffusionPipeline, PipelineOutput
 
 class BagelPipeline(DiffusionPipeline):
     model_cpu_offload_seq = "bagel_model"
@@ -7034,12 +7033,12 @@ class BagelPipeline(DiffusionPipeline):
             **infer_kwargs,
         )
 
-        out_kwargs: Dict[str, Any] = {}
+        out: Dict[str, Any] = {}
         if result.get("image") is not None:
-            out_kwargs["images"] = [result["image"]]
+            out["images"] = [result["image"]]
         if result.get("text") is not None:
-            out_kwargs["text"] = result["text"]
-        return PipelineOutput(**out_kwargs)
+            out["text"] = result["text"]
+        return out
 
     def to(self, device):
         super().to(device)          # moves registered modules