Add Marqo/dunzhang-stella_en_400M_v5

src/marqo_commons/model_registry/model_properties_data/hf_properties.py

@@ -25,6 +25,7 @@ class HFModelProperties(ModelProperties):
     notes: str = ""
     text_query_prefix: str = ""
     text_chunk_prefix: str = ""
+    trustRemoteCode: bool = False
 
     @classmethod
     def get_all_model_properties_objects(cls) -> Dict[str, "HFModelProperties"]:
@@ -246,6 +247,13 @@ def get_all_model_properties_objects(cls) -> Dict[str, "HFModelProperties"]:
                 tokens=512,
                 text_query_prefix="Represent this sentence for searching relevant passages: ",
             ),
+            "Marqo/dunzhang-stella_en_400M_v5": HFModelProperties(
+                name="Marqo/dunzhang-stella_en_400M_v5",
+                dimensions=1024,
+                tokens=512,
+                type=ModelType.hf_stella,
+                trustRemoteCode=True,
+            ),
         }
         pass
 

src/marqo_commons/model_registry/model_properties_object.py

@@ -23,6 +23,7 @@ class ModelProperties(BaseModel, ABC):
     vector_numeric_type: VectorNumericType = Field(
         ..., title="Model vector numeric type"
     )
+    trustRemoteCode: bool = Field(default=False, title="Trust remote code")
 
     def __init__(self, **kwargs):
         if "memory_size" not in kwargs:

src/marqo_commons/shared_utils/enums.py

@@ -21,6 +21,7 @@ class ModelType(str, Enum):
     test = "test"
     no_model = "no_model"
     languagebind = "languagebind"
+    hf_stella = "hf_stella"
 
 
 class VectorNumericType(str, Enum):