make a WA in case Bert model do not have safetensor file (#515)

Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>

Author: kaixuanliu

backends/python/server/text_embeddings_server/models/__init__.py

@@ -64,7 +64,15 @@ def get_model(model_path: Path, dtype: Optional[str], pool: str):
                 return DefaultModel(
                     model_path, device, datatype, pool, trust_remote=TRUST_REMOTE_CODE
                 )
-            return FlashBert(model_path, device, datatype)
+            try:
+                return FlashBert(model_path, device, datatype)
+            except FileNotFoundError as e:
+                logger.info(
+                    "Do not have safetensors file for this model, use default transformers model path instead"
+                )
+                return DefaultModel(
+                    model_path, device, datatype, pool, trust_remote=TRUST_REMOTE_CODE
+                )
         if config.architectures[0].endswith("Classification"):
             return ClassificationModel(
                 model_path, device, datatype, trust_remote=TRUST_REMOTE_CODE