diff --git a/requirements.txt b/requirements.txt index 1289318b..6434b10d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,8 +2,10 @@ coremltools>=8.0 diffusers[torch]==0.30.2 diffusionkit==0.4.0 torch -transformers==4.44.2 +transformers==4.45.2 scipy +numpy<1.24; python_version < "3.12" +numpy>=1.26,<3; python_version >= "3.12" scikit-learn pytest invisible-watermark diff --git a/setup.py b/setup.py index 644fa456..a1c62f94 100644 --- a/setup.py +++ b/setup.py @@ -17,10 +17,11 @@ "coremltools>=8.0", "diffusers[torch]==0.30.2", "torch", - "transformers==4.44.2", + "transformers==4.45.2", "huggingface-hub==0.24.6", "scipy", - "numpy<1.24", + "numpy<1.24; python_version < '3.12'", + "numpy>=1.26,<3; python_version >= '3.12'", "pytest", "scikit-learn", "invisible-watermark", @@ -29,14 +30,17 @@ "diffusionkit==0.4.0", ], packages=find_packages(), + python_requires=">=3.8", classifiers=[ "Development Status :: 4 - Beta", "Intended Audience :: Developers", "Operating System :: MacOS :: MacOS X", "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3.7", "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", + "Programming Language :: Python :: 3.10", + "Programming Language :: Python :: 3.11", + "Programming Language :: Python :: 3.12", "Topic :: Artificial Intelligence", "Topic :: Scientific/Engineering", "Topic :: Software Development",