flake/packages/rwkv/default.nix
Max 079f83ba45 Major Upgrade/Refactor
- KoboldAI is no longer upstream maintained, so is now deprecated in
  nixified.ai

- InvokeAI: v2.3.1.post2 -> v3.3.0post3
- textgen: init at v1.7
- treewide: update flake inputs including nixpkgs
- treewide: add a bunch of new dependencies and upgrade old ones
2023-10-19 22:13:37 +01:00

32 lines
588 B
Nix

{ lib
, buildPythonPackage
, fetchPypi
, setuptools
, tokenizers
}:
buildPythonPackage rec {
pname = "rwkv";
version = "0.7.4";
format = "pyproject";
src = fetchPypi {
inherit pname version;
hash = "sha256-35hoK+o0xE+Pcc3V7G/+c8rOpQL1Xwj3JbAU3oIHM+Y=";
};
propagatedBuildInputs = [
setuptools
tokenizers
];
pythonImportsCheck = [ "rwkv" ];
meta = with lib; {
description = "The RWKV Language Model";
homepage = "https://github.com/BlinkDL/ChatRWKV";
license = licenses.asl20;
maintainers = with maintainers; [ jpetrucciani ];
};
}