yzong12138
commited on
Commit
•
57f68e5
1
Parent(s):
8c8ced6
Add model
Browse files- README.md +4 -5
- definition.json +1 -1
README.md
CHANGED
@@ -17,12 +17,11 @@ The model can be loaded with [experimaestro
|
|
17 |
IR](https://experimaestro-ir.readthedocs.io/en/latest/)
|
18 |
|
19 |
```py from xpmir.models import AutoModel
|
|
|
|
|
20 |
|
21 |
-
#
|
22 |
-
AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE")
|
23 |
-
|
24 |
-
# Use this if you want to actually use the model model =
|
25 |
-
AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE", as_instance=True)
|
26 |
model.initialize() model.rsv("walgreens store sales average", "The average
|
27 |
Walgreens salary ranges...")
|
28 |
```
|
|
|
17 |
IR](https://experimaestro-ir.readthedocs.io/en/latest/)
|
18 |
|
19 |
```py from xpmir.models import AutoModel
|
20 |
+
# Model that can be re-used in experiments
|
21 |
+
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE")
|
22 |
|
23 |
+
# Use this if you want to actually use the model
|
24 |
+
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE", as_instance=True)
|
|
|
|
|
|
|
25 |
model.initialize() model.rsv("walgreens store sales average", "The average
|
26 |
Walgreens salary ranges...")
|
27 |
```
|
definition.json
CHANGED
@@ -1 +1 @@
|
|
1 |
-
[{"id":
|
|
|
1 |
+
[{"id": 140014190408992, "module": "xpmir.text.huggingface", "type": "TransformerTokensEncoder", "typename": "xpmir.text.huggingface.TransformerTokensEncoder", "identifier": "d26c7c4d00bb2002f07f80036294ed81355c1256fde3758c6ab9451c7fc06c2c", "fields": {"model_id": "distilbert-base-uncased", "trainable": true, "layer": 0, "dropout": 0.0}}, {"id": 140014190407744, "module": "xpmir.neural.splade", "type": "MaxAggregation", "typename": "xpmir.neural.splade.MaxAggregation", "identifier": "6d85ac05decb4f139354ac3f79990a2afdb8ced21c8861f2dd5c720484514f48", "fields": {}}, {"id": 140014190409328, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "typename": "xpmir.neural.splade.SpladeTextEncoder", "identifier": "9f455ba2eb9c3dc5ce40d4f0620b24a8d086b1a35449380af6cf625a69f2efeb", "fields": {"encoder": {"type": "python", "value": 140014190408992}, "aggregation": {"type": "python", "value": 140014190407744}, "maxlen": 200}}, {"id": 140014190411296, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "typename": "xpmir.neural.splade.SpladeTextEncoder", "identifier": "af02fa8dcdc136f3c02eeb124fe524302c85d736deeaa6d04b3acc215d1593ac", "fields": {"encoder": {"type": "python", "value": 140014190408992}, "aggregation": {"type": "python", "value": 140014190407744}, "maxlen": 30}}, {"id": 140014190623472, "module": "xpmir.neural.dual", "type": "DotDense", "typename": "xpmir.neural.dual.DotDense", "identifier": "d4e16bd0471b8b48c996614c52a927c911296acf20fca61b75b431499cf47b47", "fields": {"checkpoint": {"type": "path.serialized", "value": "checkpoint", "is_folder": true}, "encoder": {"type": "python", "value": 140014190409328}, "query_encoder": {"type": "python", "value": 140014190411296}}}]
|