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

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