mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-09 13:57:54 -05:00
remove shark 1.0 tests, add support for 2.0 llm
* add support for external weights * add tests and edit deps
This commit is contained in:
7
setup.py
7
setup.py
@@ -9,11 +9,6 @@ with open("README.md", "r", encoding="utf-8") as fh:
|
||||
|
||||
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.5"
|
||||
backend_deps = []
|
||||
if "NO_BACKEND" in os.environ.keys():
|
||||
backend_deps = [
|
||||
"iree-compiler>=20221022.190",
|
||||
"iree-runtime>=20221022.190",
|
||||
]
|
||||
|
||||
setup(
|
||||
name="nodai-SHARK",
|
||||
@@ -39,7 +34,5 @@ setup(
|
||||
install_requires=[
|
||||
"numpy",
|
||||
"PyYAML",
|
||||
"torch-mlir",
|
||||
]
|
||||
+ backend_deps,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user