Add "safe_serialization": true as is required for custom Nomic models
Browse filesHello!
As discussed, this adds safe_serialization=True when loading the model. I haven't tested this personally, but I have a strong suspicion that it should work, based on [this](https://huggingface.co/nomic-ai/nomic-embed-text-v1.5/blob/main/sentence_bert_config.json#L5).
- Tom Aarsen
sentence_bert_config.json
CHANGED
@@ -2,6 +2,7 @@
|
|
2 |
"max_seq_length": 8192,
|
3 |
"do_lower_case": false,
|
4 |
"model_args": {
|
5 |
-
"add_pooling_layer": false
|
|
|
6 |
}
|
7 |
}
|
|
|
2 |
"max_seq_length": 8192,
|
3 |
"do_lower_case": false,
|
4 |
"model_args": {
|
5 |
+
"add_pooling_layer": false,
|
6 |
+
"safe_serialization": true
|
7 |
}
|
8 |
}
|