Skip to content

Commit 0add749

Browse files
committed
format
1 parent 415ff40 commit 0add749

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

optimum/exporters/onnx/__main__.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -410,10 +410,7 @@ def main_export(
410410
**loading_kwargs,
411411
)
412412

413-
needs_pad_token_id = (
414-
task == "text-classification"
415-
and getattr(model.config, "pad_token_id", None) is None
416-
)
413+
needs_pad_token_id = task == "text-classification" and getattr(model.config, "pad_token_id", None) is None
417414

418415
if needs_pad_token_id:
419416
if pad_token_id is not None:

0 commit comments

Comments
 (0)