Update aggregate.py

aggregate.py

@@ -119,7 +119,7 @@ class BatchAggregator:
             )
             return pipeline(
                 "text2text-generation",
-                model_name,
+                model=model_name,
                 device=self.device,
                 torch_dtype=torch.float32,
             )