nixpkgs/pkgs/by-name/ex/exo/package.nix
2024-11-15 14:41:40 +01:00

82 lines
1.5 KiB
Nix

{
lib,
stdenv,
fetchFromGitHub,
python3Packages,
unstableGitUpdater,
}:
python3Packages.buildPythonApplication {
pname = "exo";
version = "0-unstable-2024-10-29";
pyproject = true;
src = fetchFromGitHub {
owner = "exo-explore";
repo = "exo";
rev = "50a1b171f64a75594793147aa39db933ef38aed9";
hash = "sha256-hzguPEQDTKBWe8Um+pwJpsctPbQqA6QW5LGknKWFaKo=";
};
build-system = with python3Packages; [ setuptools ];
pythonRelaxDeps = true;
pythonRemoveDeps = [ "uuid" ];
dependencies = with python3Packages; [
aiohttp
aiohttp-cors
aiofiles
blobfile
grpcio
grpcio-tools
jinja2
netifaces
numpy
nvidia-ml-py
pillow
prometheus-client
protobuf
psutil
pydantic
requests
rich
safetensors
tenacity
tqdm
transformers
tinygrad
];
pythonImportsCheck = [
"exo"
"exo.inference.tinygrad.models"
];
nativeCheckInputs = with python3Packages; [
mlx
pytestCheckHook
];
disabledTestPaths = [
"test/test_tokenizers.py"
];
# Tests require `mlx` which is not supported on linux.
doCheck = stdenv.isDarwin;
passthru = {
updateScript = unstableGitUpdater {
hardcodeZeroVersion = true;
};
};
meta = {
description = "Run your own AI cluster at home with everyday devices";
homepage = "https://github.com/exo-explore/exo";
license = lib.licenses.gpl3Only;
maintainers = with lib.maintainers; [ GaetanLepage ];
mainProgram = "exo";
};
}