Merge pull request #258134 from mweinelt/k-diffusion-0.1.0
python310Packages.k-diffusion: 0.0.16 -> 0.1.0; python310Packages.dctorch: init at 0.1.2; python310Packages.rotary-embedded-torch: init at 0.3.0
This commit is contained in:
commit
a2d7b3ebe4
42
pkgs/development/python-modules/dctorch/default.nix
Normal file
42
pkgs/development/python-modules/dctorch/default.nix
Normal file
|
@ -0,0 +1,42 @@
|
||||||
|
{ lib
|
||||||
|
, buildPythonPackage
|
||||||
|
, fetchPypi
|
||||||
|
, poetry-core
|
||||||
|
, numpy
|
||||||
|
, scipy
|
||||||
|
, torch
|
||||||
|
}:
|
||||||
|
|
||||||
|
buildPythonPackage rec {
|
||||||
|
pname = "dctorch";
|
||||||
|
version = "0.1.2";
|
||||||
|
pyproject = true;
|
||||||
|
|
||||||
|
src = fetchPypi {
|
||||||
|
inherit pname version;
|
||||||
|
hash = "sha256-TmfLAkiofrQNWYBhIlY4zafbZPgFftISCGloO/rlEG4=";
|
||||||
|
};
|
||||||
|
|
||||||
|
nativeBuildInputs = [
|
||||||
|
poetry-core
|
||||||
|
];
|
||||||
|
|
||||||
|
propagatedBuildInputs = [
|
||||||
|
numpy
|
||||||
|
scipy
|
||||||
|
torch
|
||||||
|
];
|
||||||
|
|
||||||
|
pythonImportsCheck = [
|
||||||
|
"dctorch"
|
||||||
|
];
|
||||||
|
|
||||||
|
doCheck = false; # no tests
|
||||||
|
|
||||||
|
meta = with lib; {
|
||||||
|
description = "Fast discrete cosine transforms for pytorch";
|
||||||
|
homepage = "https://pypi.org/project/dctorch/";
|
||||||
|
license = licenses.mit;
|
||||||
|
maintainers = teams.tts.members;
|
||||||
|
};
|
||||||
|
}
|
|
@ -3,13 +3,15 @@
|
||||||
, buildPythonPackage
|
, buildPythonPackage
|
||||||
, clean-fid
|
, clean-fid
|
||||||
, clip-anytorch
|
, clip-anytorch
|
||||||
|
, dctorch
|
||||||
, einops
|
, einops
|
||||||
, fetchFromGitHub
|
, fetchFromGitHub
|
||||||
, jsonmerge
|
, jsonmerge
|
||||||
, kornia
|
, kornia
|
||||||
, pillow
|
, pillow
|
||||||
, pythonOlder
|
, pythonOlder
|
||||||
, resize-right
|
, rotary-embedding-torch
|
||||||
|
, safetensors
|
||||||
, scikit-image
|
, scikit-image
|
||||||
, scipy
|
, scipy
|
||||||
, torch
|
, torch
|
||||||
|
@ -22,7 +24,7 @@
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "k-diffusion";
|
pname = "k-diffusion";
|
||||||
version = "0.0.16";
|
version = "0.1.0";
|
||||||
format = "setuptools";
|
format = "setuptools";
|
||||||
|
|
||||||
disabled = pythonOlder "3.7";
|
disabled = pythonOlder "3.7";
|
||||||
|
@ -31,20 +33,22 @@ buildPythonPackage rec {
|
||||||
owner = "crowsonkb";
|
owner = "crowsonkb";
|
||||||
repo = "k-diffusion";
|
repo = "k-diffusion";
|
||||||
rev = "refs/tags/v${version}";
|
rev = "refs/tags/v${version}";
|
||||||
hash = "sha256-tOWDFt0/hGZF5HENiHPb9a2pBlXdSvDvCNTsCMZljC4=";
|
hash = "sha256-jcIA0HfEnVHk9XDXPevGBw81GsXlm1Ztp8ceNirShEA=";
|
||||||
};
|
};
|
||||||
|
|
||||||
propagatedBuildInputs = [
|
propagatedBuildInputs = [
|
||||||
accelerate
|
accelerate
|
||||||
clean-fid
|
clean-fid
|
||||||
clip-anytorch
|
clip-anytorch
|
||||||
|
dctorch
|
||||||
einops
|
einops
|
||||||
jsonmerge
|
jsonmerge
|
||||||
kornia
|
kornia
|
||||||
pillow
|
pillow
|
||||||
resize-right
|
rotary-embedding-torch
|
||||||
scikit-image
|
scikit-image
|
||||||
scipy
|
scipy
|
||||||
|
safetensors
|
||||||
torch
|
torch
|
||||||
torchdiffeq
|
torchdiffeq
|
||||||
torchsde
|
torchsde
|
||||||
|
|
|
@ -0,0 +1,48 @@
|
||||||
|
{ lib
|
||||||
|
, buildPythonPackage
|
||||||
|
, fetchFromGitHub
|
||||||
|
|
||||||
|
# build-system
|
||||||
|
, setuptools
|
||||||
|
, wheel
|
||||||
|
|
||||||
|
# dependencies
|
||||||
|
, einops
|
||||||
|
, torch
|
||||||
|
}:
|
||||||
|
|
||||||
|
buildPythonPackage rec {
|
||||||
|
pname = "rotary-embedding-torch";
|
||||||
|
version = "0.3.0";
|
||||||
|
pyproject = true;
|
||||||
|
|
||||||
|
src = fetchFromGitHub {
|
||||||
|
owner = "lucidrains";
|
||||||
|
repo = "rotary-embedding-torch";
|
||||||
|
rev = version;
|
||||||
|
hash = "sha256-fGyBBPfvVq1iZ2m2NNjmHSK+iy76N/09Pt11YDyOyN4=";
|
||||||
|
};
|
||||||
|
|
||||||
|
nativeBuildInputs = [
|
||||||
|
setuptools
|
||||||
|
wheel
|
||||||
|
];
|
||||||
|
|
||||||
|
propagatedBuildInputs = [
|
||||||
|
einops
|
||||||
|
torch
|
||||||
|
];
|
||||||
|
|
||||||
|
pythonImportsCheck = [
|
||||||
|
"rotary_embedding_torch"
|
||||||
|
];
|
||||||
|
|
||||||
|
doCheck = false; # no tests
|
||||||
|
|
||||||
|
meta = with lib; {
|
||||||
|
description = "Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch";
|
||||||
|
homepage = "https://github.com/lucidrains/rotary-embedding-torch";
|
||||||
|
license = licenses.mit;
|
||||||
|
maintainers = teams.tts.members;
|
||||||
|
};
|
||||||
|
}
|
|
@ -2613,6 +2613,8 @@ self: super: with self; {
|
||||||
|
|
||||||
dcmstack = callPackage ../development/python-modules/dcmstack { };
|
dcmstack = callPackage ../development/python-modules/dcmstack { };
|
||||||
|
|
||||||
|
dctorch = callPackage ../development/python-modules/dctorch { };
|
||||||
|
|
||||||
ddt = callPackage ../development/python-modules/ddt { };
|
ddt = callPackage ../development/python-modules/ddt { };
|
||||||
|
|
||||||
deal = callPackage ../development/python-modules/deal { };
|
deal = callPackage ../development/python-modules/deal { };
|
||||||
|
@ -11519,6 +11521,8 @@ self: super: with self; {
|
||||||
|
|
||||||
ropper = callPackage ../development/python-modules/ropper { };
|
ropper = callPackage ../development/python-modules/ropper { };
|
||||||
|
|
||||||
|
rotary-embedding-torch = callPackage ../development/python-modules/rotary-embedding-torch { };
|
||||||
|
|
||||||
rouge-score = callPackage ../development/python-modules/rouge-score { };
|
rouge-score = callPackage ../development/python-modules/rouge-score { };
|
||||||
|
|
||||||
routeros-api = callPackage ../development/python-modules/routeros-api { };
|
routeros-api = callPackage ../development/python-modules/routeros-api { };
|
||||||
|
|
Loading…
Reference in a new issue