Merge remote-tracking branch 'upstream/ean-unify-sd' into ean-unify-sd

.github/workflows/test_models.yml

@@ -47,8 +47,10 @@ jobs:
           source turbine_venv/bin/activate
           python3.11 -m pip install --upgrade pip
           pip install --no-compile -r ${{ github.workspace }}/iree-turbine/pytorch-cpu-requirements.txt
-          pip install --no-compile --pre --no-cache-dir -r ${{ github.workspace }}/iree-turbine/iree-requirements.txt -f https://iree.dev/pip-release-links.html
-          pip install --no-compile --pre --upgrade -e models -r models/requirements.txt 
+          pip install --pre --upgrade -r ${{ github.workspace }}/iree-turbine/requirements.txt
+          pip install --no-compile --pre -e ${{ github.workspace }}/iree-turbine[testing]
+          pip install --upgrade --pre --no-cache-dir iree-compiler iree-runtime -f https://iree.dev/pip-release-links.html
+          pip install --no-compile --pre --upgrade -e models -r models/requirements.txt
       
       - name: Show current free memory
         run: |

.gitignore

@@ -28,4 +28,7 @@ wheelhouse
 *.safetensors
 *.gguf
 *.vmfb
-*.mlir
+*.mlir
+*.npy
+*.png
+*tmp*

models/requirements.txt

@@ -4,7 +4,8 @@ shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
 transformers==4.37.1
 torchsde
 accelerate
-diffusers @ git+https://github.com/nod-ai/[email protected]
+peft
+diffusers @ git+https://github.com/nod-ai/[email protected]
 brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
 # turbine tank downloading/uploading
 azure-storage-blob

models/setup.py

@@ -57,9 +57,9 @@ def load_version_info():
         "Shark-Turbine",
         "protobuf",
         "sentencepiece",
-        "transformers==4.37.1",
+        "transformers>=4.37.1",
         "accelerate",
-        "diffusers==0.24.0",
+        "diffusers==0.29.0.dev0",
         "azure-storage-blob",
         "einops",
     ],