Add support for nomic-ai/nomic-embed-text-v1.5 model

optimum/commands/env.py

@@ -57,8 +57,8 @@ def run(self):
             "Platform": platform.platform(),
             "Python version": platform.python_version(),
             "Huggingface_hub version": huggingface_hub.__version__,
-            "PyTorch version (GPU?)": f"{pt_version} (cuda availabe: {pt_cuda_available})",
-            "Tensorflow version (GPU?)": f"{tf_version} (cuda availabe: {tf_cuda_available})",
+            "PyTorch version (GPU?)": f"{pt_version} (cuda available: {pt_cuda_available})",
+            "Tensorflow version (GPU?)": f"{tf_version} (cuda available: {tf_cuda_available})",
         }
 
         print("\nCopy-and-paste the text below in your GitHub issue:\n")

optimum/exporters/onnx/model_configs.py

@@ -115,6 +115,10 @@ def inputs(self) -> Dict[str, Dict[int, str]]:
         }
 
 
+class NomicBertOnnxConfig(BertOnnxConfig):
+    DEFAULT_ONNX_OPSET = 14
+
+
 class AlbertOnnxConfig(BertOnnxConfig):
     DEFAULT_ONNX_OPSET = 11
 

optimum/exporters/tasks.py

@@ -826,6 +826,7 @@ class TasksManager:
             "text2text-generation-with-past",
             onnx="M2M100OnnxConfig",
         ),
+        "nomic-bert": supported_tasks_mapping("feature-extraction", onnx="NomicBertOnnxConfig"),
         "nystromformer": supported_tasks_mapping(
             "feature-extraction",
             "fill-mask",

optimum/utils/normalized_config.py

@@ -247,6 +247,7 @@ class NormalizedConfigManager:
         "mpt": MPTNormalizedTextConfig,
         "mt5": T5LikeNormalizedTextConfig,
         "m2m-100": BartLikeNormalizedTextConfig,
+        "nomic-bert": NormalizedTextConfig,
         "nystromformer": NormalizedTextConfig,
         "opt": NormalizedTextConfig,
         "pegasus": BartLikeNormalizedTextConfig,