Upload ZeroShotEmbedding

config.json

@@ -12,5 +12,5 @@
   "model_type": "embedding-head",
   "output_size": 128,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.0"
+  "transformers_version": "4.35.1"
 }

model.py

@@ -88,6 +88,14 @@ class ZeroShotEmbeddingForClustering(PreTrainedModel):
                               prompted_embeddings.transpose(0, 1))
         return similarity
 
+    @classmethod
+    def from_pretrained_base(cls, pretrained_model_name_or_path):
+        head_model = ZeroShotEmbedding.from_pretrained(
+            pretrained_model_name_or_path)
+        model = cls(head_model.config)
+        cls.head_model = head_model
+        return model
+
 
 ZeroShotEmbeddingConfig.register_for_auto_class()
 ZeroShotEmbedding.register_for_auto_class("AutoModel")

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5c190260213a40bd005edc34138ac93485adc7386575a5a8e2d5ec5ecd2479c0
+oid sha256:427af19366ad65a03d77feeab3e7dc1d8c4a217efd3fe97052724ea94c3ca2d0
 size 13640544