From f3be817d3026aee9c962746a2a827e5e63bda926 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 23 May 2024 17:02:49 +0200 Subject: [PATCH] update setup --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 9d54b5d9c9..0057c34c90 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ INSTALL_REQUIRE = [ "torch>=1.11", "transformers>=4.36.0,<4.42.0", - "optimum @ git+https://github.com/huggingface/optimum.git@fix-bt-ci", + "optimum~=1.19", "datasets>=1.4.0", "sentencepiece", "scipy",