diff options
| -rw-r--r-- | .SRCINFO | 35 | ||||
| -rw-r--r-- | PKGBUILD | 49 |
2 files changed, 84 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..d44ac3bf7e77 --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,35 @@ +pkgbase = python-transformer-debugger + pkgdesc = Tool for investigations into specific behaviors of small language models + pkgver = 0.0.0 + pkgrel = 1 + url = https://github.com/openai/transformer-debugger + arch = any + groups = openai + license = Apache + checkdepends = mypy + checkdepends = python-pytest + checkdepends = python-pytest-asyncio + makedepends = python-build + makedepends = python-installer + makedepends = python-setuptools + makedepends = python-wheel + depends = pre-commit + depends = python-aiohttp + depends = python-blobfile + depends = python-boostedblob + depends = python-click + depends = python-fastapi + depends = python-fire + depends = python-httpx + depends = python-numpy + depends = python-orjson + depends = python-pydantic + depends = python-scikit-learn + depends = python-starlette + depends = python-tiktoken + depends = python-torch + depends = python-uvicorn + source = python-transformer-debugger-0.0.0.tar.gz::https://github.com/openai/transformer-debugger/archive/refs/heads/main.tar.gz + sha256sums = 46de1870ff1ca77f1d6d5187ab58c0169e39da0164e126948aec75e20eee265d + +pkgname = python-transformer-debugger diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..e0a7a577328b --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,49 @@ +# Maintainer: Daniel Bershatsky <bepshatsky@yandex.ru> + +pkgname=python-transformer-debugger +_pkgname=${pkgname#python-} +pkgver=0.0.0 +pkgrel=1 +pkgdesc='Tool for investigations into specific behaviors of small language models' +arch=('any') +url='https://github.com/openai/transformer-debugger' +license=('Apache') +groups=('openai') +depends=( + 'pre-commit' + 'python-aiohttp' + 'python-blobfile' + 'python-boostedblob' + 'python-click' + 'python-fastapi' + 'python-fire' + 'python-httpx' + 'python-numpy' + 'python-orjson' + 'python-pydantic' + 'python-scikit-learn' + 'python-starlette' + 'python-tiktoken' + 'python-torch' + 'python-uvicorn' +) +checkdepends=( + 'mypy' + 'python-pytest' + 'python-pytest-asyncio' +) +makedepends=('python-build' 'python-installer' 'python-setuptools' 'python-wheel') +optdepends=() +source=("$pkgname-0.0.0.tar.gz::$url/archive/refs/heads/main.tar.gz") +sha256sums=('46de1870ff1ca77f1d6d5187ab58c0169e39da0164e126948aec75e20eee265d') + +build() { + python -m build -nw $srcdir/$_pkgname-main +} + +package() { + python -m installer \ + --compile-bytecode 1 \ + --destdir $pkgdir \ + $srcdir/$_pkgname-main/dist/neuron_explainer-0.0.1-py3-none-any.whl +} |