update _sanitize_parameters to accept `generate_kwargs`

pipeline.py

@@ -66,12 +66,13 @@ class QASRL_Pipeline(Text2TextGenerationPipeline):
         
     def _sanitize_parameters(self, **kwargs):
         preprocess_kwargs, forward_kwargs, postprocess_kwargs = {}, {}, {} # super()._sanitize_parameters(**kwargs)
-        if "predicate_marker" in kwargs:
-            preprocess_kwargs["predicate_marker"] = kwargs["predicate_marker"]
-        if "predicate_type" in kwargs:
-            preprocess_kwargs["predicate_type"] = kwargs["predicate_type"]
-        if "verb_form" in kwargs:
-            preprocess_kwargs["verb_form"] = kwargs["verb_form"]
+        forward_kwargs.update(kwargs.get("generate_kwargs", dict()))
+        forward_kwargs.update(kwargs.get("model_kwargs", dict()))
+        preprocess_keywords = ("predicate_marker", "predicate_type", "verb_form")
+        for key in preprocess_keywords:
+            if key in kwargs:
+                preprocess_kwargs[key] = kwargs[key]
+
         return preprocess_kwargs, forward_kwargs, postprocess_kwargs
 
     def preprocess(self, inputs, predicate_marker="<predicate>", predicate_type=None, verb_form=None):