Upload ZeroShotEmbedding

model.py

@@ -17,6 +17,8 @@ class ZeroShotEmbeddingConfig(PretrainedConfig):
 
 
 class ZeroShotEmbedding(PreTrainedModel):
+    config_class = ZeroShotEmbeddingConfig
+
     def __init__(self, config):
         super(ZeroShotEmbedding, self).__init__(config)
 
@@ -66,6 +68,8 @@ class ZeroShotEmbedding(PreTrainedModel):
 
 
 class ZeroShotEmbeddingForClustering(PreTrainedModel):
+    config_class = ZeroShotEmbeddingConfig
+
     def __init__(self, config):
         super(ZeroShotEmbeddingForClustering, self).__init__(config)
         self.base_embedding_model = SentenceTransformer(

model.safetensors

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