Remove flair from supported models

nlpodyssey · Oct 30, 2023 · 4ca06be · 4ca06be
1 parent 200646a
commit 4ca06be
Show file tree

Hide file tree

Showing 47 changed files with 2 additions and 3,867 deletions.
diff --git a/pkg/converter/converter.go b/pkg/converter/converter.go
@@ -6,11 +6,8 @@ package converter
 
 import (
 	"fmt"
-	"strings"
-
 	"github.com/nlpodyssey/cybertron/pkg/converter/bart"
 	"github.com/nlpodyssey/cybertron/pkg/converter/bert"
-	"github.com/nlpodyssey/cybertron/pkg/converter/flair"
 	"github.com/nlpodyssey/cybertron/pkg/models"
 	"github.com/nlpodyssey/spago/mat/float"
 )
@@ -31,19 +28,12 @@ func Convert[T float.DType](modelPath string, overwriteIfExists bool) error {
 		return bert.Convert[T](modelPath, overwriteIfExists)
 	case "bart", "marian", "pegasus":
 		return bart.Convert[T](modelPath, overwriteIfExists)
-	case "flair":
-		return flair.Convert[T](modelPath, overwriteIfExists)
 	default:
 		return fmt.Errorf("unsupported model type: %#v", modelType)
 	}
 }
 
 func resolveModelType(modelPath string) (string, error) {
-	if strings.Contains(modelPath, "flair") {
-		// Handling the case where there is no configuration file
-		return "flair", nil
-	}
-
 	config, err := models.ReadCommonModelConfig(modelPath, "")
 	if err != nil {
 		return "", err

diff --git a/pkg/converter/flair/conversion/builtins/getattr.go b/pkg/converter/flair/conversion/builtins/getattr.go
diff --git a/pkg/converter/flair/conversion/builtins/int.go b/pkg/converter/flair/conversion/builtins/int.go
diff --git a/pkg/converter/flair/conversion/collections/defaultdict.go b/pkg/converter/flair/conversion/collections/defaultdict.go
diff --git a/pkg/converter/flair/conversion/flair/crf.go b/pkg/converter/flair/conversion/flair/crf.go
diff --git a/pkg/converter/flair/conversion/flair/dictionary.go b/pkg/converter/flair/conversion/flair/dictionary.go
diff --git a/pkg/converter/flair/conversion/flair/flairembeddings.go b/pkg/converter/flair/conversion/flair/flairembeddings.go