Pin versions for important depedencies

Author: TobyRoseman

requirements.txt

@@ -1,8 +1,8 @@
-coremltools>=7.0
-diffusers[torch]
-diffusionkit
+coremltools>=8.0
+diffusers[torch]==0.30.2
+diffusionkit==0.4.0
 torch
-transformers==4.29.2
+transformers==4.44.2
 scipy
 scikit-learn
 pytest

setup.py

@@ -14,19 +14,19 @@
     long_description_content_type='text/markdown',
     author='Apple Inc.',
     install_requires=[
-        "coremltools>=7.0b2",
-        "diffusers[torch]",
+        "coremltools>=8.0",
+        "diffusers[torch]==0.30.2",
         "torch",
-        "transformers>=4.30.0",
-        "huggingface-hub",
+        "transformers==4.44.2",
+        "huggingface-hub==0.24.6",
         "scipy",
         "numpy<1.24",
         "pytest",
         "scikit-learn",
         "invisible-watermark",
         "safetensors",
         "matplotlib",
-        "diffusionkit",
+        "diffusionkit==0.4.0",
     ],
     packages=find_packages(),
     classifiers=[