Don't assume that all Flair models have WordEmbeddings

marco-nicola · marco-nicola · commit 64b4552f796c · 2022-09-18T17:22:42.000+02:00
diff --git a/pkg/converter/flair/convert.go b/pkg/converter/flair/convert.go
@@ -193,8 +193,6 @@ func (conv *converter[T]) encoderEmbeddingsTokensEncoder() ([]flair.TokensEncode
 	}
 
 	switch {
-	case flairWe == nil:
-		return nil, fmt.Errorf("WordEmbeddings not found in StackedEmbeddings")
 	case flairEmbForward == nil:
 		return nil, fmt.Errorf("FlairEmbeddings-forward not found in StackedEmbeddings")
 	case flairEmbBackward == nil:

Original file line number	Diff line number	Diff line change
`@@ -193,8 +193,6 @@ func (conv *converter[T]) encoderEmbeddingsTokensEncoder() ([]flair.TokensEncode`
`193`	`193`	`}`
`194`	`194`
`195`	`195`	`switch {`
`196`		`- case flairWe == nil:`
`197`		`- return nil, fmt.Errorf("WordEmbeddings not found in StackedEmbeddings")`
`198`	`196`	`case flairEmbForward == nil:`
`199`	`197`	`return nil, fmt.Errorf("FlairEmbeddings-forward not found in StackedEmbeddings")`
`200`	`198`	`case flairEmbBackward == nil:`