diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py index f42cd8908a..ad04f193d7 100644 --- a/optimum/intel/ipex/modeling_base.py +++ b/optimum/intel/ipex/modeling_base.py @@ -134,6 +134,7 @@ class IPEXModel(OptimizedModel): main_input_name = "input_ids" output_name = "last_hidden_state" _supports_cache_class = False + _is_stateful = False def __init__( self, diff --git a/optimum/intel/version.py b/optimum/intel/version.py index e2caf8ae11..ad90088f0c 100644 --- a/optimum/intel/version.py +++ b/optimum/intel/version.py @@ -12,4 +12,4 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "1.18.1" +__version__ = "1.18.2" diff --git a/setup.py b/setup.py index 853d012e54..da10f4f91b 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ INSTALL_REQUIRE = [ "torch>=1.11", "transformers>=4.36.0,<4.44.0", - "optimum@git+https://github.com/huggingface/optimum.git@v1.21.3-release", + "optimum>=1.21.3,<1.22.0", "datasets>=1.4.0", "sentencepiece", "setuptools",