Update model_config.yaml
Browse files- model_config.yaml +2 -2
model_config.yaml
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
name:
|
2 |
config_type: model
|
3 |
-
task:
|
4 |
activation: gelu
|
5 |
attention_dropout: 0.1
|
6 |
dim: 768
|
|
|
1 |
+
name: distilbert_mask_filling
|
2 |
config_type: model
|
3 |
+
task: mask_filling
|
4 |
activation: gelu
|
5 |
attention_dropout: 0.1
|
6 |
dim: 768
|