Merge pull request #232238 from mweinelt/tts-0.14.0 (67b3dc9b) · Commits · nix / nixpkgs

pkgs/development/python-modules/clean-fid/default.nix

0 → 100644

+50 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchFromGitHub

		# dependencies
		, numpy
		, pillow
		, requests
		, scipy
		, torch
		, torchvision
		, tqdm
		}:

		buildPythonPackage rec {
		pname = "clean-fid";
		version = "0.1.35";
		format = "setuptools";

		src = fetchFromGitHub {
		owner = "GaParmar";
		repo = "clean-fid";
		rev = "c8ffa420a3923e8fd87c1e75170de2cf59d2644b";
		hash = "sha256-fqBU/TmCXDTPU3KTP0+VYQoP+HsT2UMcZeLzQHKD9hw=";
		};

		propagatedBuildInputs = [
		numpy
		pillow
		requests
		scipy
		torch
		torchvision
		tqdm
		];

		pythonImportsCheck = [
		"cleanfid"
		];

		# no tests1
		doCheck = false;

		meta = with lib; {
		description = "PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]";
		homepage = "https://github.com/GaParmar/clean-fid";
		license = licenses.mit;
		maintainers = teams.tts.members;
		};
		}

0 → 100644

+57 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchFromGitHub

		# dependencies
		, ftfy
		, regex
		, tqdm
		, torch
		, torchvision

		# tests
		, pytestCheckHook
		}:

		buildPythonPackage rec {
		pname = "clip-anytorch";
		version = "2.5.2";
		format = "setuptools";

		src = fetchFromGitHub {
		owner = "rom1504";
		repo = "CLIP";
		rev = version;
		hash = "sha256-EqVkpMQHawoCFHNupf49NrvLdGCq35wnYBpdP81Ztd4=";
		};

		propagatedBuildInputs = [
		ftfy
		regex
		tqdm
		torch
		torchvision
		];

		pythonImportsCheck = [
		"clip"
		];

		# all tests require network access
		doCheck = false;

		nativeCheckInputs = [
		pytestCheckHook
		];

		preCheck = ''
		export HOME=$TMPDIR
		'';

		meta = with lib; {
		description = "Contrastive Language-Image Pretraining";
		homepage = "https://github.com/rom1504/CLIP";
		license = licenses.mit;
		maintainers = teams.tts.members;
		};
		}

0 → 100644

+69 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchFromGitHub

		# dependencies
		, accelerate
		, clean-fid
		, clip-anytorch
		, einops
		, jsonmerge
		, kornia
		, pillow
		, resize-right
		, scikit-image
		, scipy
		, torch
		, torchdiffeq
		, torchsde
		, torchvision
		, tqdm
		, wandb

		}:

		buildPythonPackage rec {
		pname = "k-diffusion";
		version = "0.0.14";
		format = "setuptools";

		src = fetchFromGitHub {
		owner = "crowsonkb";
		repo = "k-diffusion";
		rev = "v${version}";
		hash = "sha256-KKVgk+1hidDBVaRnXjoqwuSRydI10OPHK3YModAizZU=";
		};

		propagatedBuildInputs = [
		accelerate
		clean-fid
		clip-anytorch
		einops
		jsonmerge
		kornia
		pillow
		resize-right
		scikit-image
		scipy
		torch
		torchdiffeq
		torchsde
		torchvision
		tqdm
		wandb
		];

		pythonImportsCheck = [
		"k_diffusion"
		];

		# no tests
		doCheck = false;

		meta = with lib; {
		description = "Karras et al. (2022) diffusion models for PyTorch";
		homepage = "https://github.com/crowsonkb/k-diffusion";
		license = licenses.mit;
		maintainers = teams.tts.members;
		};
		}

0 → 100644

+37 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchPypi

		# dependencies
		, numpy
		, torch
		}:

		buildPythonPackage rec {
		pname = "resize-right";
		version = "0.0.2";

		src = fetchPypi {
		inherit pname version;
		hash = "sha256-fcNbcs5AErd/fMkEmDUWN5OrmKWKuIk2EPsRn+Wa9SA=";
		};

		propagatedBuildInputs = [
		numpy
		torch
		];

		pythonImportsCheck = [
		"resize_right"
		];

		# no tests
		doCheck = false;

		meta = with lib; {
		description = "The correct way to resize images or tensors. For Numpy or Pytorch (differentiable";
		homepage = "https://github.com/assafshocher/ResizeRight";
		license = licenses.mit;
		maintainers = teams.tts.members;
		};
		}

0 → 100644

+43 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchPypi

		# dependencies
		, torch
		, scipy

		# tests
		, pytestCheckHook
		}:

		buildPythonPackage rec {
		pname = "torchdiffeq";
		version = "0.2.3";
		format = "setuptools";

		src = fetchPypi {
		inherit pname version;
		hash = "sha256-/nX0NLkJCsDCdwLgK+0hRysPhwNb5lgfUe3F1AE+oxo=";
		};

		propagatedBuildInputs = [
		torch
		scipy
		];

		pythonImportsCheck = [ "torchdiffeq" ];

		# no tests in sdist, no tags on git
		doCheck = false;

		nativeCheckInputs = [
		pytestCheckHook
		];

		meta = with lib; {
		description = "Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation";
		homepage = "https://github.com/rtqichen/torchdiffeq";
		license = licenses.mit;
		maintainers = teams.tts.members;
		};
		}