naver
/

provence-reranker-debertav3-v1

Model card Files Files and versions Community

nadiinchi commited on Jan 15

Commit

bbb1f20

·

verified ·

1 Parent(s): 27108b5

fix title bug

Files changed (1) hide show

modeling_provence.py +1 -1

modeling_provence.py CHANGED Viewed

@@ -119,7 +119,7 @@ class Provence(DebertaV2PreTrainedModel):
             contexts = [[context]]
         else:
             contexts = context
-        if type(title) == str:
             titles = [[title]]
         else:
             titles = title

             contexts = [[context]]
         else:
             contexts = context
+        if type(title) == str and title != "first_sentence":
             titles = [[title]]
         else:
             titles = title