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:
Daniel Garvey
2023-12-14 21:44:37 -06:00
committed by GitHub
parent f692a012e1
commit ebfcfec338
16 changed files with 377 additions and 576 deletions

View File

@@ -5,14 +5,25 @@ requires = [
"packaging",
"numpy>=1.22.4",
"torch-mlir>=20230620.875",
"iree-compiler>=20221022.190",
"iree-runtime>=20221022.190",
]
build-backend = "setuptools.build_meta"
[tool.black]
line-length = 79
include = '\.pyi?$'
exclude = "apps/language_models/scripts/vicuna.py"
extend-exclude = "apps/language_models/src/pipelines/minigpt4_pipeline.py"
exclude = '''
(
/(
| apps/stable_diffusion
| apps/language_models
| shark
| benchmarks
| tank
| build
| generated_imgs
| shark.venv
)/
| setup.py
)
'''