diff --git a/pkgs/development/python-modules/clip-anytorch/default.nix b/pkgs/development/python-modules/clip-anytorch/default.nix new file mode 100644 index 00000000000..b689d5708d5 --- /dev/null +++ b/pkgs/development/python-modules/clip-anytorch/default.nix @@ -0,0 +1,57 @@ +{ 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; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index fc37a0252fb..315383a90ed 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1843,6 +1843,8 @@ self: super: with self; { clip = callPackage ../development/python-modules/clip { }; + clip-anytorch = callPackage ../development/python-modules/clip-anytorch { }; + cock = callPackage ../development/python-modules/cock { }; click = callPackage ../development/python-modules/click { };