Spaces:

anamargarida
/

Trial

Sleeping

anamargarida commited on Mar 11

Commit

aeb0521

verified ·

1 Parent(s): 0179253

Update modeling_st2.py

Files changed (1) hide show

modeling_st2.py CHANGED Viewed

@@ -213,7 +213,6 @@ class ST2ModelV2(nn.Module):
         start_effect_logits,
         end_cause_logits,
         end_effect_logits,
-        word_ids,
         topk=5
     ):
         # basic post processing (removing logits from [CLS], [SEP], [PAD])

         start_effect_logits,
         end_cause_logits,
         end_effect_logits,
         topk=5
     ):
         # basic post processing (removing logits from [CLS], [SEP], [PAD])