Upload MBartForConditionalGeneration
Browse files- README.md +42 -0
- adapter_config.json +21 -0
- head_config.json +15 -0
- pytorch_adapter.bin +3 -0
- pytorch_model_head.bin +3 -0
README.md
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
tags:
|
3 |
+
- mbart
|
4 |
+
- adapter-transformers
|
5 |
+
datasets:
|
6 |
+
- ETR-fr
|
7 |
+
---
|
8 |
+
|
9 |
+
# Adapter `koena/mbart-large-50-prefix-ETR-fr` for facebook/mbart-large-50
|
10 |
+
|
11 |
+
An [adapter](https://adapterhub.ml) for the `facebook/mbart-large-50` model that was trained on the [ETR-fr](https://huggingface.co/datasets/ETR-fr/) dataset.
|
12 |
+
|
13 |
+
This adapter was created for usage with the **[Adapters](https://github.com/Adapter-Hub/adapters)** library.
|
14 |
+
|
15 |
+
## Usage
|
16 |
+
|
17 |
+
First, install `adapters`:
|
18 |
+
|
19 |
+
```
|
20 |
+
pip install -U adapters
|
21 |
+
```
|
22 |
+
|
23 |
+
Now, the adapter can be loaded and activated like this:
|
24 |
+
|
25 |
+
```python
|
26 |
+
from adapters import AutoAdapterModel
|
27 |
+
|
28 |
+
model = AutoAdapterModel.from_pretrained("facebook/mbart-large-50")
|
29 |
+
adapter_name = model.load_adapter("koena/mbart-large-50-prefix-ETR-fr", set_active=True)
|
30 |
+
```
|
31 |
+
|
32 |
+
## Architecture & Training
|
33 |
+
|
34 |
+
<!-- Add some description here -->
|
35 |
+
|
36 |
+
## Evaluation results
|
37 |
+
|
38 |
+
<!-- Add some description here -->
|
39 |
+
|
40 |
+
## Citation
|
41 |
+
|
42 |
+
<!-- Add some description here -->
|
adapter_config.json
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"config": {
|
3 |
+
"architecture": "prefix_tuning",
|
4 |
+
"bottleneck_size": 512,
|
5 |
+
"cross_prefix": true,
|
6 |
+
"dropout": 0.0,
|
7 |
+
"encoder_prefix": true,
|
8 |
+
"flat": false,
|
9 |
+
"leave_out": [],
|
10 |
+
"non_linearity": "tanh",
|
11 |
+
"prefix_length": 150,
|
12 |
+
"shared_gating": true,
|
13 |
+
"use_gating": false
|
14 |
+
},
|
15 |
+
"hidden_size": 1024,
|
16 |
+
"model_class": "MBartForConditionalGeneration",
|
17 |
+
"model_name": "facebook/mbart-large-50",
|
18 |
+
"model_type": "mbart",
|
19 |
+
"name": "ETR-fr",
|
20 |
+
"version": "adapters.1.0.0"
|
21 |
+
}
|
head_config.json
ADDED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"config": null,
|
3 |
+
"hidden_size": 1024,
|
4 |
+
"label2id": {
|
5 |
+
"LABEL_0": 0,
|
6 |
+
"LABEL_1": 1,
|
7 |
+
"LABEL_2": 2
|
8 |
+
},
|
9 |
+
"model_class": "MBartForConditionalGeneration",
|
10 |
+
"model_name": "facebook/mbart-large-50",
|
11 |
+
"model_type": "mbart",
|
12 |
+
"name": null,
|
13 |
+
"num_labels": 3,
|
14 |
+
"version": "adapters.1.0.0"
|
15 |
+
}
|
pytorch_adapter.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7faa5572863332d2bde3f3cc003b4a53ef67cf37bca7e00c34fde30c292e7245
|
3 |
+
size 159536065
|
pytorch_model_head.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec5032c29c4029fde3bdc0559b6a9d559c399955c51e88bc1c23e3184d3fe270
|
3 |
+
size 1025222938
|