From a054b028d5b54fcbb36b14e5e605fc12cefb9c50 Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Tue, 7 Jan 2025 14:58:50 +0000 Subject: [PATCH] Code formatting --- optimum/exporters/onnx/model_configs.py | 2 +- optimum/utils/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index 30f62eace3..d6cd965e16 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -34,6 +34,7 @@ DummyInputGenerator, DummyIntGenerator, DummyPastKeyValuesGenerator, + DummyPatchTSTInputGenerator, DummyPix2StructInputGenerator, DummyPointsGenerator, DummySeq2SeqDecoderTextInputGenerator, @@ -52,7 +53,6 @@ GemmaDummyPastKeyValuesGenerator, GPTBigCodeDummyPastKeyValuesGenerator, MistralDummyPastKeyValuesGenerator, - DummyPatchTSTInputGenerator, NormalizedConfig, NormalizedEncoderDecoderConfig, NormalizedSeq2SeqConfig, diff --git a/optimum/utils/__init__.py b/optimum/utils/__init__.py index 34610fb999..d1ef410e79 100644 --- a/optimum/utils/__init__.py +++ b/optimum/utils/__init__.py @@ -61,6 +61,7 @@ DummyIntGenerator, DummyLabelsGenerator, DummyPastKeyValuesGenerator, + DummyPatchTSTInputGenerator, DummyPix2StructInputGenerator, DummyPointsGenerator, DummySeq2SeqDecoderTextInputGenerator, @@ -80,7 +81,6 @@ GPTBigCodeDummyPastKeyValuesGenerator, MistralDummyPastKeyValuesGenerator, MultiQueryPastKeyValuesGenerator, - DummyPatchTSTInputGenerator, ) from .modeling_utils import recurse_getattr, recurse_setattr from .normalized_config import (