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

   1 { buildPythonPackage
   2 , fetchFromGitHub
   3 , ftfy
   4 , lib
   5 , regex
   6 , torch
   7 , torchvision
   8 , tqdm
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "clip";
  13   version = "unstable-2022-11-17";
  14   format = "setuptools";
  15
  16   src = fetchFromGitHub {
  17     owner = "openai";
  18     repo = pname;
  19     rev = "d50d76daa670286dd6cacf3bcd80b5e4823fc8e1";
  20     hash = "sha256-GAitNBb5CzFVv2+Dky0VqSdrFIpKKtoAoyqeLoDaHO4=";
  21   };
  22
  23   propagatedBuildInputs = [
  24     ftfy
  25     regex
  26     torch
  27     torchvision
  28     tqdm
  29   ];
  30
  31   pythonImportsCheck = [ "clip" ];
  32
  33   meta = with lib; {
  34     description = "Contrastive Language-Image Pretraining";
  35     homepage = "https://github.com/openai/CLIP";
  36     license = licenses.mit;
  37     maintainers = with maintainers; [ samuela ];
  38   };
  39 }