mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
cb509343d9
- fix setup_venv.sh for benchmarks/imports etc. - fix torch benchmarks in SharkBenchmarkRunner - generate SD artifacts using build_tools/stable_diffusion_testing.py and --import_mlir - decouple SD gen from tank/generate_sharktank for now
19 lines
422 B
TOML
19 lines
422 B
TOML
[build-system]
|
|
requires = [
|
|
"setuptools>=42",
|
|
"wheel",
|
|
"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"
|