Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.0 kB · 98 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899{ lib, buildPythonPackage, fetchFromGitHub,
# build-system pdm-backend,
# dependencies ftfy, huggingface-hub, protobuf, regex, safetensors, sentencepiece, timm, torch, torchvision, tqdm,
# checks pytestCheckHook, braceexpand, pandas, transformers, webdataset,
stdenv,}:buildPythonPackage rec { pname = "open-clip-torch"; version = "2.32.0"; pyproject = true;
src = fetchFromGitHub { owner = "mlfoundations"; repo = "open_clip"; tag = "v${version}"; hash = "sha256-HXzorEAVPieCHfW3xzXqNTTIzJSbIuaZhcfcp0htdCk="; };
build-system = [ pdm-backend ];
dependencies = [ ftfy huggingface-hub protobuf regex safetensors sentencepiece timm torch torchvision tqdm ];
nativeCheckInputs = [ pytestCheckHook braceexpand pandas transformers webdataset ];
pythonImportsCheck = [ "open_clip" ];
# -> On Darwin: # AttributeError: Can't pickle local object 'build_params.<locals>.<lambda>' # -> On Linux: # KeyError: Caught KeyError in DataLoader worker process 0 disabledTestPaths = [ "tests/test_wds.py" ];
disabledTests = [ # requires network "test_download_pretrained_from_hfh" "test_inference_simple" "test_inference_with_data" "test_pretrained_text_encoder" "test_training_mt5" # fails due to type errors "test_num_shards" ] ++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [ "test_training" "test_training_coca" "test_training_unfreezing_vit" "test_training_clip_with_jit" ];
meta = { description = "Open source implementation of CLIP"; homepage = "https://github.com/mlfoundations/open_clip"; changelog = "https://github.com/mlfoundations/open_clip/releases/tag/${src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ iynaix ]; mainProgram = "open-clip"; };}