From 14002068446c35c69b6d64a0cffff19e918e6a84 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 30 Jan 2025 11:30:18 +0100 Subject: [PATCH] set optimum version to 1.24 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 663528ef9c..cf4bfb3191 100644 --- a/setup.py +++ b/setup.py @@ -28,7 +28,7 @@ INSTALL_REQUIRE = [ "torch>=1.11", - "optimum@git+https://github.com/huggingface/optimum.git", + "optimum~=1.24", "transformers>=4.36,<4.48", "datasets>=1.4.0", "sentencepiece",