Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Merged
merged 12 commits into from
Jul 10, 2024
Merged

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
ade5b3a
Adds one SD pipeline to rule them all
monorimet Jul 2, 2024
c2d8d5f
Initial punet integration.
monorimet Jul 2, 2024
a9ead64
Formatting.
monorimet Jul 2, 2024
0147cc9
Partitioned Unet I8 support.
monorimet Jul 8, 2024
4bc57c2
Fixes for non-punet attn spec
monorimet Jul 8, 2024
34d3d84
Fixes to punet, device IDs
eagarvey-amd Jul 9, 2024
86f3a80
Add TODO to encode comment in VAE export script.
eagarvey-amd Jul 9, 2024
4c01167
Test fixes, reduce unet model signature variance
eagarvey-amd Jul 10, 2024
22eb78c
Update test_shark.yml
monorimet Jul 10, 2024
4d8960d
Use hip driver unless rocm-legacy is used as device string.
eagarvey-amd Jul 10, 2024
e8de325
Switch device args to hip in CI and xfail prompt encoder test for now.
eagarvey-amd Jul 10, 2024
708b8d0
Address comments
eagarvey-amd Jul 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/test_shark.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
4 changes: 3 additions & 1 deletion models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
protobuf
shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
gguf
transformers==4.37.1
torchsde
accelerate
Expand All @@ -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
Loading
Loading