nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-30 02:42:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

python310Packages.clip-anytorch: init at 2.5.2

Browse Source
This commit is contained in:
Martin Weinelt 2023-05-16 14:01:28 +02:00
parent a1a072ee66
commit 1b28a9b9b8
No known key found for this signature in database
GPG Key ID: 87C1E9888F856759
2 changed files with 59 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
pkgs/development/python-modules/clip-anytorch/default.nix Normal file
View File

@ -0,0 +1,57 @@
{ 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;
};
}

2
pkgs/top-level/python-packages.nix
View File

@ -1843,6 +1843,8 @@ self: super: with self; {
clip = callPackage ../development/python-modules/clip { };
clip-anytorch = callPackage ../development/python-modules/clip-anytorch { };
cock = callPackage ../development/python-modules/cock { };
click = callPackage ../development/python-modules/click { };