[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
fbff0292bd
commit
b46269f7dd
1 changed files with 1 additions and 1 deletions
|
|
@ -787,7 +787,7 @@ class FastBaseModel:
|
|||
num_labels = kwargs.pop("num_labels", None)
|
||||
id2label = kwargs.pop("id2label", None)
|
||||
label2id = kwargs.pop("label2id", None)
|
||||
|
||||
|
||||
if num_labels is not None:
|
||||
model_config.num_labels = num_labels
|
||||
if id2label is not None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue