From 73d93bdd938f6663e6acda85d33d0e6fe0ac081a Mon Sep 17 00:00:00 2001 From: Thanaji Rao Thakkalapelli Date: Wed, 28 Aug 2024 08:45:44 -0700 Subject: [PATCH] Downgrade sympy to match synapaseAI 1.18 base image (#215) --- server/requirements.txt | 2 +- server/text_generation_server/habana_quantization_env.py | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/server/requirements.txt b/server/requirements.txt index 55e734046..304aa33ae 100644 --- a/server/requirements.txt +++ b/server/requirements.txt @@ -70,7 +70,7 @@ sentence-transformers[train]==3.0.1 ; python_version >= "3.9" and python_version sentencepiece==0.1.99 ; python_version >= "3.9" and python_version < "3.13" setuptools==73.0.1 ; python_version >= "3.9" and python_version < "3.13" six==1.16.0 ; python_version >= "3.9" and python_version < "3.13" -sympy==1.13.2 ; python_version >= "3.9" and python_version < "3.13" +sympy==1.12.1 ; python_version >= "3.9" and python_version < "3.13" threadpoolctl==3.5.0 ; python_version >= "3.9" and python_version < "3.13" tokenizers==0.19.1 ; python_version >= "3.9" and python_version < "3.13" tqdm==4.66.5 ; python_version >= "3.9" and python_version < "3.13" diff --git a/server/text_generation_server/habana_quantization_env.py b/server/text_generation_server/habana_quantization_env.py index a854e45aa..3c06fd098 100644 --- a/server/text_generation_server/habana_quantization_env.py +++ b/server/text_generation_server/habana_quantization_env.py @@ -1,9 +1,6 @@ # Copyright (C) 2024 Habana Labs, Ltd. an Intel Company. import os -import sys - -assert "habana_frameworks" not in sys.modules quant_config = os.getenv("QUANT_CONFIG", "") is_quantization_enabled = quant_config != ""