Skip to content

Commit 3803b00

Browse files
don't save config and tokenizer data as instance parameters
1 parent 272b8e1 commit 3803b00

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

flair/embeddings/base.py

+2
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,8 @@ def __init__(
190190
**kwargs,
191191
):
192192
self.instance_parameters = self.get_instance_parameters(locals=locals())
193+
del self.instance_parameters["saved_config"]
194+
del self.instance_parameters["tokenizer_data"]
193195
super().__init__()
194196
# temporary fix to disable tokenizer parallelism warning
195197
# (see https://stackoverflow.com/questions/62691279/how-to-disable-tokenizers-parallelism-true-false-warning)

flair/embeddings/document.py

-1
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ def __init__(
4141
:param layer_mean: If True, uses a scalar mix of layers as embedding
4242
:param fine_tune: If True, allows transformers to be fine-tuned during training
4343
"""
44-
super().__init__()
4544
TransformerEmbedding.__init__(
4645
self,
4746
model=model,

flair/embeddings/token.py

-1
Original file line numberDiff line numberDiff line change
@@ -901,7 +901,6 @@ def __init__(
901901
:param layer_mean: If True, uses a scalar mix of layers as embedding
902902
:param fine_tune: If True, allows transformers to be fine-tuned during training
903903
"""
904-
super().__init__()
905904
TransformerEmbedding.__init__(
906905
self, model=model, is_token_embedding=True, is_document_embedding=is_document_embedding, **kwargs
907906
)

0 commit comments

Comments
 (0)