47 lines
864 B
Nix
47 lines
864 B
Nix
|
{ lib
|
||
|
|
, buildPythonPackage
|
||
|
|
, fetchFromGitHub
|
||
|
|
, flax
|
||
|
|
, jax
|
||
|
|
, jaxlib
|
||
|
|
, transformers
|
||
|
|
}:
|
||
|
|
|
||
|
|
buildPythonPackage rec {
|
||
|
|
pname = "vqgan-jax";
|
||
|
|
version = "unstable-2022-04-20";
|
||
|
|
|
||
|
|
src = fetchFromGitHub {
|
||
|
|
owner = "patil-suraj";
|
||
|
|
repo = "vqgan-jax";
|
||
|
|
rev = "1be20eee476e5d35c30e4ec3ed12222018af8ce4";
|
||
|
|
sha256 = "sha256-OZihAXpE0UsgauQ38XDmAF+lrIgz05uK0ro8SCdVsPc=";
|
||
|
|
};
|
||
|
|
|
||
|
|
format = "setuptools";
|
||
|
|
|
||
|
|
buildInputs = [
|
||
|
|
jaxlib
|
||
|
|
];
|
||
|
|
|
||
|
|
propagatedBuildInputs = [
|
||
|
|
flax
|
||
|
|
jax
|
||
|
|
transformers
|
||
|
|
];
|
||
|
|
|
||
|
|
doCheck = false;
|
||
|
|
|
||
|
|
pythonImportsCheck = [
|
||
|
|
"vqgan_jax"
|
||
|
|
];
|
||
|
|
|
||
|
|
meta = with lib; {
|
||
|
|
description = "JAX implementation of VQGAN";
|
||
|
|
homepage = "https://github.com/patil-suraj/vqgan-jax";
|
||
|
|
# license unknown: https://github.com/patil-suraj/vqgan-jax/issues/9
|
||
|
|
license = lib.licenses.unfree;
|
||
|
|
maintainers = with maintainers; [ r-burns ];
|
||
|
|
};
|
||
|
|
}
|