param-bharat
/

ModernBERT-large-nli-scorer

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

param-bharat commited on Dec 27, 2024

Commit

8d23c68

·

verified ·

1 Parent(s): c3dbbe7

Upload NLIScorer

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -365,7 +365,7 @@ class NLIScorer(Pipeline):
         postprocess_kwargs = {}
         if "task_type" in kwargs:
             preprocess_kwargs["task_type"] = kwargs["task_type"]
-            preprocess_kwargs["task_type"] = kwargs["task_type"]
         return preprocess_kwargs, {}, postprocess_kwargs
     def preprocess(self, inputs, task_type):

         postprocess_kwargs = {}
         if "task_type" in kwargs:
             preprocess_kwargs["task_type"] = kwargs["task_type"]
+            postprocess_kwargs["task_type"] = kwargs["task_type"]
         return preprocess_kwargs, {}, postprocess_kwargs
     def preprocess(self, inputs, task_type):