Skip to content

Commit c79029b

Browse files
committed
fix(configspace): Fix flair configspace
1 parent 6d18471 commit c79029b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

embeddings/config/flair_config_space.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
class FlairTextClassificationConfigSpaceMapping:
1919
LOAD_MODEL_KEYS_MAPPING: ClassVar[Mapping[str, Set[str]]] = MappingProxyType(
2020
{
21-
"FlairDocumentCNNEmbeddings": {
21+
"FlairDocumentRNNEmbeddings": {
2222
"hidden_size",
2323
"rnn_type",
2424
"rnn_layers",
@@ -27,7 +27,7 @@ class FlairTextClassificationConfigSpaceMapping:
2727
"word_dropout",
2828
"reproject_words",
2929
},
30-
"FlairDocumentRNNEmbeddings": {
30+
"FlairDocumentCNNEmbeddings": {
3131
"cnn_pool_kernels",
3232
"dropout",
3333
"word_dropout",

0 commit comments

Comments
 (0)