pyrox.dev
1{ buildPythonPackage
2, fetchFromGitHub
3, ftfy
4, lib
5, regex
6, torch
7, torchvision
8, tqdm
9}:
10
11buildPythonPackage 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}