Pin torch-mlir and diffusers.

pyproject.toml

@@ -5,7 +5,7 @@ requires = [
     "packaging",
 
     "numpy>=1.22.4",
-    "torch-mlir>=20230620.875",
+    "torch-mlir==20231210.*",
     "iree-compiler==20231212.*",
     "iree-runtime==20231212.*",
 ]

requirements.txt

@@ -21,7 +21,7 @@ parameterized
 # Add transformers, diffusers and scipy since it most commonly used
 tokenizers==0.13.3
 transformers
-diffusers
+diffusers==0.24.0
 #accelerate is now required for diffusers import from ckpt.
 accelerate
 scipy

setup_venv.ps1

@@ -89,7 +89,7 @@ else {python -m venv .\shark1.venv\}
 python -m pip install --upgrade pip
 pip install wheel
 pip install -r requirements.txt
-pip install --pre torch-mlir torchvision torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://llvm.github.io/torch-mlir/package-index/
+pip install --pre torch-mlir==20231210.* torchvision torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://llvm.github.io/torch-mlir/package-index/
 pip install --upgrade -f https://nod-ai.github.io/SRT/pip-release-links.html iree-compiler==20231212.* iree-runtime==20231212.*
 Write-Host "Building SHARK..."
 pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SRT/pip-release-links.html