Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit 5eb827d

Browse files
pre-commit-ci[bot]changwangss
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d70106d commit 5eb827d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intel_extension_for_transformers/transformers/modeling/modeling_auto.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1274,7 +1274,7 @@ def train_func(model):
12741274
break
12751275
output = model(**inputs)
12761276
if isinstance(output, tuple):
1277-
loss = output[0].mean()
1277+
loss = output[0].mean()
12781278
elif isinstance(output, dict):
12791279
loss = output["logits"].mean()
12801280
else:

0 commit comments

Comments
 (0)