From 399bcb05c160d34ee097d2002f59314b4a2132dd Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Wed, 30 Aug 2023 21:13:43 -0700 Subject: [PATCH 1/2] using dict update --- text2text/pytorch_pretrained_bert/modeling.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/text2text/pytorch_pretrained_bert/modeling.py b/text2text/pytorch_pretrained_bert/modeling.py index b57d6b2..1524f3f 100755 --- a/text2text/pytorch_pretrained_bert/modeling.py +++ b/text2text/pytorch_pretrained_bert/modeling.py @@ -135,8 +135,7 @@ def __init__(self, def from_dict(cls, json_object): """Constructs a `BertConfig` from a Python dictionary of parameters.""" config = BertConfig(vocab_size_or_config_json_file=-1) - for key, value in json_object.items(): - config.__dict__[key] = value + config.__dict__.update(json_object) return config @classmethod From ffb04f1be5e71ddf04d5f277455eacb8307fd89a Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Wed, 30 Aug 2023 21:16:01 -0700 Subject: [PATCH 2/2] use dict.update --- text2text/pytorch_pretrained_bert/modeling.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/text2text/pytorch_pretrained_bert/modeling.py b/text2text/pytorch_pretrained_bert/modeling.py index 1524f3f..1e3f218 100755 --- a/text2text/pytorch_pretrained_bert/modeling.py +++ b/text2text/pytorch_pretrained_bert/modeling.py @@ -105,8 +105,7 @@ def __init__(self, if isinstance(vocab_size_or_config_json_file, str): with open(vocab_size_or_config_json_file, "r", encoding='utf-8') as reader: json_config = json.loads(reader.read()) - for key, value in json_config.items(): - self.__dict__[key] = value + self.__dict__.update(json_config) elif isinstance(vocab_size_or_config_json_file, int): self.vocab_size = vocab_size_or_config_json_file self.hidden_size = hidden_size