token-classification pipeline added

maxdotio · Mar 2, 2022 · 967cca0 · 967cca0
1 parent 9e3628d
commit 967cca0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",
 }