Skip to content

Commit

Permalink
Fix installing and importing iree-turbine (#2175)
Browse files Browse the repository at this point in the history
The package was renamed from `shark-turbine` to `iree.turbine` with
iree-org/iree-turbine@40016ad.
  • Loading branch information
marbre authored Oct 18, 2024
1 parent 6b79037 commit 5fef4f6
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion apps/shark_studio/modules/ckpt_processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import requests
import torch
import safetensors
from shark_turbine.aot.params import (
from iree.turbine.aot.params import (
ParameterArchiveBuilder,
)
from io import BytesIO
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ wheel


torch==2.3.0
shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
iree-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
turbine-models @ git+https://github.com/nod-ai/SHARK-ModelDev.git@main#subdirectory=models
diffusers @ git+https://github.com/nod-ai/[email protected]
brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
Expand Down

0 comments on commit 5fef4f6

Please sign in to comment.