ROCm-safe bfloat16 inference with custom matmul attention. Automatic output directories, random seed, timing metadata.
25 lines
410 B
TOML
25 lines
410 B
TOML
[build-system]
|
|
requires = ["setuptools>=64"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "vnassets"
|
|
version = "0.1.0"
|
|
requires-python = ">=3.12"
|
|
dependencies = [
|
|
"torch",
|
|
"diffusers",
|
|
"transformers",
|
|
"accelerate",
|
|
"safetensors",
|
|
"pillow",
|
|
"pyyaml",
|
|
"click",
|
|
]
|
|
|
|
[tool.setuptools.packages.find]
|
|
exclude = ["models*"]
|
|
|
|
[project.scripts]
|
|
vnasset = "vnassets.cli:main"
|