riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Adds type checking to PreTrainedConfig. (#20926)

Browse Source
This commit is contained in:
Matthew McDermott 2022-12-30 02:35:01 -05:00 committed by GitHub
parent 8637316e5e
commit 491a33d138
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/configuration_utils.py
View File

@ -732,7 +732,7 @@ class PretrainedConfig(PushToHubMixin):
return json.loads(text)
def __eq__(self, other):
return self.__dict__ == other.__dict__
return isinstance(other, PretrainedConfig) and (self.__dict__ == other.__dict__)
def __repr__(self):
return f"{self.__class__.__name__} {self.to_json_string()}"