mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-01-09 22:45:08 +00:00
79 lines
1.4 KiB
Nix
79 lines
1.4 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
fetchFromGitHub,
|
|
python3Packages,
|
|
}:
|
|
python3Packages.buildPythonApplication {
|
|
pname = "exo";
|
|
version = "0-unstable-2024-10-21";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "exo-explore";
|
|
repo = "exo";
|
|
rev = "82a708f974b9a720e127c38a383f22e129be6373";
|
|
hash = "sha256-BhhcYOipdLAviTzWRdNLMMPiND4mYv9Mkn8/yxo0vXY=";
|
|
};
|
|
|
|
build-system = with python3Packages; [ setuptools ];
|
|
|
|
pythonRelaxDeps = true;
|
|
|
|
pythonRemoveDeps = [ "uuid" ];
|
|
|
|
dependencies = with python3Packages; [
|
|
aiohttp
|
|
aiohttp-cors
|
|
aiofiles
|
|
blobfile
|
|
grpcio
|
|
grpcio-tools
|
|
hf-transfer
|
|
huggingface-hub
|
|
jinja2
|
|
netifaces
|
|
numpy
|
|
nvidia-ml-py
|
|
pillow
|
|
prometheus-client
|
|
protobuf
|
|
psutil
|
|
requests
|
|
rich
|
|
safetensors
|
|
tailscale
|
|
tenacity
|
|
tiktoken
|
|
tokenizers
|
|
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;
|
|
|
|
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";
|
|
};
|
|
}
|