Unverified Commit a2d7b3eb authored by OTABI Tomoya's avatar OTABI Tomoya Committed by GitHub
Browse files

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
parents ee3bc8bf d385f4fb

pkgs/development/python-modules/dctorch/default.nix

0 → 100644
+42 −0
Original line number Diff line number Diff line 
{ 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;

  };

}

pkgs/development/python-modules/k-diffusion/default.nix

+8 −4
Original line number Diff line number Diff line
@@ -3,13 +3,15 @@ 
, buildPythonPackage

, clean-fid

, clip-anytorch

, dctorch

, einops

, fetchFromGitHub

, jsonmerge

, kornia

, pillow

, pythonOlder

, resize-right

, rotary-embedding-torch

, safetensors

, scikit-image

, scipy

, torch
@@ -22,7 +24,7 @@ 


buildPythonPackage rec {

  pname = "k-diffusion";

  version = "0.0.16";

  version = "0.1.0";

  format = "setuptools";



  disabled = pythonOlder "3.7";
@@ -31,20 +33,22 @@ buildPythonPackage rec { 
    owner = "crowsonkb";

    repo = "k-diffusion";

    rev = "refs/tags/v${version}";

    hash = "sha256-tOWDFt0/hGZF5HENiHPb9a2pBlXdSvDvCNTsCMZljC4=";

    hash = "sha256-jcIA0HfEnVHk9XDXPevGBw81GsXlm1Ztp8ceNirShEA=";

  };



  propagatedBuildInputs = [

    accelerate

    clean-fid

    clip-anytorch

    dctorch

    einops

    jsonmerge

    kornia

    pillow

    resize-right

    rotary-embedding-torch

    scikit-image

    scipy

    safetensors

    torch

    torchdiffeq

    torchsde

pkgs/development/python-modules/rotary-embedding-torch/default.nix

0 → 100644
+48 −0
Original line number Diff line number Diff line 
{ 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;

  };

}

pkgs/top-level/python-packages.nix

+4 −0
Original line number Diff line number Diff line
@@ -2613,6 +2613,8 @@ self: super: with self; { 














  dcmstack = callPackage ../development/python-modules/dcmstack { };



























  dctorch = callPackage ../development/python-modules/dctorch { };



























  ddt = callPackage ../development/python-modules/ddt { };





























  deal = callPackage ../development/python-modules/deal { };










@@ -11519,6 +11521,8 @@ self: super: with self; {



























  ropper = callPackage ../development/python-modules/ropper { };



























  rotary-embedding-torch = callPackage ../development/python-modules/rotary-embedding-torch { };



























  rouge-score = callPackage ../development/python-modules/rouge-score { };





























  routeros-api = callPackage ../development/python-modules/routeros-api { };