mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-25 07:05:13 +00:00
82 lines
1.5 KiB
Nix
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";
|
|
};
|
|
}
|