Consolidates SD pipelines and adds support for sharktank unet. (#766)

nod-ai · Jul 10, 2024 · d534cd4 · d534cd4
1 parent 79a094f
commit d534cd4
Show file tree

Hide file tree

Showing 27 changed files with 1,939 additions and 1,922 deletions.
diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
@@ -20,7 +20,7 @@ jobs:
     strategy:
       matrix:
         version: [3.11]
-        os: [nodai-ubuntu-builder-large]
+        os: [nodai-amdgpu-mi250-x86-64]
 
     runs-on: ${{matrix.os}}
     steps:

diff --git a/models/requirements.txt b/models/requirements.txt
@@ -1,5 +1,5 @@
 protobuf
-shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
+gguf
 transformers==4.37.1
 torchsde
 accelerate
@@ -12,3 +12,5 @@ azure-storage-blob
 einops
 pytest
 scipy
+shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
+sharktank @ git+https://github.com/nod-ai/sharktank@main#subdirectory=sharktank