Migrate Sentence Transformer namespace and ZM class names

README.md

@@ -34,14 +34,14 @@ import os
 os.environ["KERAS_BACKEND"] = "torch"  # set before importing Keras / ZeroModels
 
 from keras import ops
-from zeromodels.st_models import (
-    SentenceTransformer,
-    SentenceTransformerTokenizer,
+from zeromodels.sentence_transformers import (
+    ZMSentenceTransformer,
+    ZMSentenceTransformerTokenizer,
 )
 
 repo = "zeromodels/all-mpnet-base-v2"
-model = SentenceTransformer.from_weights(repo)
-tokenizer = SentenceTransformerTokenizer.from_weights(repo)
+model = ZMSentenceTransformer.from_weights(repo)
+tokenizer = ZMSentenceTransformerTokenizer.from_weights(repo)
 
 texts = ["A person is reading a book.", "Someone reads a novel."]
 embeddings = model.encode(texts, tokenizer=tokenizer)
@@ -94,7 +94,7 @@ All 13 converted variants use `from_weights("zeromodels/<variant>")`:
 - Use the tokenizer from the **same** repo as the model. It preserves the source
   truncation length of 384 tokens for this checkpoint.
 - For exact source loading without these converted Keras weights, use
-  `SentenceTransformer.from_weights("hf:sentence-transformers/all-mpnet-base-v2")`.
+  `ZMSentenceTransformer.from_weights("hf:sentence-transformers/all-mpnet-base-v2")`.
 - The repository contains `model.weights.h5`, `zm_config.json`, and
   `tokenizer.json`. The pinned source commit is listed above.
 

zm_config.json

@@ -1,8 +1,8 @@
 {
   "library_name": "zeromodels",
   "zeromodels_version": "1.3.0",
-  "model_module": "zeromodels.st_models.sentence_transformer",
-  "model_class": "SentenceTransformerModel",
+  "model_module": "zeromodels.sentence_transformers",
+  "model_class": "ZMSentenceTransformer",
   "variant": "all-mpnet-base-v2",
   "weights": "model.weights.h5",
   "schema_version": 2,

zm_preprocessor.json

@@ -1,7 +1,7 @@
 {
   "library_name": "zeromodels",
-  "preprocessor_module": "zeromodels.st_models",
-  "preprocessor_class": "SentenceTransformerTokenizer",
+  "preprocessor_module": "zeromodels.sentence_transformers",
+  "preprocessor_class": "ZMSentenceTransformerTokenizer",
   "variant": "all-mpnet-base-v2",
   "tokenizer_file": "tokenizer.json",
   "backbone_type": "mpnet",