depot/third_party/nixpkgs/pkgs/development/python-modules/clip-anytorch/default.nix
Default email 94427deb9d Project import generated by Copybara.
GitOrigin-RevId: f91ee3065de91a3531329a674a45ddcb3467a650
2023-05-24 16:37:59 +03:00

57 lines
881 B
Nix

{ lib
, buildPythonPackage
, fetchFromGitHub
# dependencies
, ftfy
, regex
, tqdm
, torch
, torchvision
# tests
, pytestCheckHook
}:
buildPythonPackage rec {
pname = "clip-anytorch";
version = "2.5.2";
format = "setuptools";
src = fetchFromGitHub {
owner = "rom1504";
repo = "CLIP";
rev = version;
hash = "sha256-EqVkpMQHawoCFHNupf49NrvLdGCq35wnYBpdP81Ztd4=";
};
propagatedBuildInputs = [
ftfy
regex
tqdm
torch
torchvision
];
pythonImportsCheck = [
"clip"
];
# all tests require network access
doCheck = false;
nativeCheckInputs = [
pytestCheckHook
];
preCheck = ''
export HOME=$TMPDIR
'';
meta = with lib; {
description = "Contrastive Language-Image Pretraining";
homepage = "https://github.com/rom1504/CLIP";
license = licenses.mit;
maintainers = teams.tts.members;
};
}