aithings/packages/openclip/default.nix
2024-02-24 19:18:13 +01:00

48 lines
882 B
Nix

# WARNING: This file was automatically generated. You should avoid editing it.
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ buildPythonPackage
, fetchFromGitHub
, ftfy
, huggingface-hub
, lib
, regex
, sentencepiece
, torch
, torchvision
, tqdm
, protobuf
}:
buildPythonPackage rec {
pname = "open_clip";
version = "2.20.0";
src = fetchFromGitHub {
owner = "mlfoundations";
repo = "open_clip";
rev = "v2.20.0";
sha256 = "sha256-Ca4oi2LqleIFAGBJB7YIi4nXe2XhOP6ErDFXgXtJLxM=";
};
propagatedBuildInputs = [
torch
torchvision
regex
ftfy
tqdm
huggingface-hub
sentencepiece
protobuf
];
# TODO FIXME
doCheck = false;
meta = with lib; {
description = "OpenCLIP";
homepage = "https://github.com/mlfoundations/open_clip";
};
}