From 967cca0bc93ef6bb50707a8610aee6a670ff534e Mon Sep 17 00:00:00 2001 From: binarymax Date: Wed, 2 Mar 2022 08:41:12 -0500 Subject: [PATCH] token-classification pipeline added --- convert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/convert.py b/convert.py index b5bf580..6b249eb 100644 --- a/convert.py +++ b/convert.py @@ -16,11 +16,11 @@ "default":"default", "sequence-classification":"sequence-classification", "question-answering":"question-answering", - "sentence-transformers":"default" + "sentence-transformers":"default", + "token-classification":"token-classification", #"masked-lm":"masked-lm", #"causal-lm":"causal-lm", #"seq2seq-lm":"seq2seq-lm", - #"token-classification":"token-classification", #"multiple-choice":"multiple-choice", }