pyrox.dev
1{
2 buildPythonPackage,
3 fetchFromGitHub,
4 ftfy,
5 lib,
6 regex,
7 torch,
8 torchvision,
9 tqdm,
10}:
11
12buildPythonPackage {
13 pname = "clip";
14 version = "unstable-2022-11-17";
15 format = "setuptools";
16
17 src = fetchFromGitHub {
18 owner = "openai";
19 repo = "clip";
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}