61 lines
1.2 KiB
Nix
61 lines
1.2 KiB
Nix
|
{
|
||
|
|
lib,
|
||
|
|
fetchFromGitHub,
|
||
|
|
python3Packages,
|
||
|
|
}:
|
||
|
|
|
||
|
|
python3Packages.buildPythonApplication rec {
|
||
|
|
pname = "skypilot";
|
||
|
|
version = "0.6.1";
|
||
|
|
|
||
|
|
src = fetchFromGitHub {
|
||
|
|
owner = "skypilot-org";
|
||
|
|
repo = "skypilot";
|
||
|
|
rev = "refs/tags/v${version}";
|
||
|
|
hash = "sha256-ZrNI9s7U39SMHqIzOtyuth8Wrkn+T2KSsMfpqO1pxoI=";
|
||
|
|
};
|
||
|
|
|
||
|
|
pyproject = true;
|
||
|
|
|
||
|
|
build-system = with python3Packages; [ setuptools ];
|
||
|
|
|
||
|
|
# when updating, please ensure package version constraints stipulaed
|
||
|
|
# in setup.py are met
|
||
|
|
propagatedBuildInputs = with python3Packages; [
|
||
|
|
cachetools
|
||
|
|
click
|
||
|
|
colorama
|
||
|
|
cryptography
|
||
|
|
filelock
|
||
|
|
jinja2
|
||
|
|
jsonschema
|
||
|
|
networkx
|
||
|
|
packaging
|
||
|
|
pandas
|
||
|
|
pendulum
|
||
|
|
prettytable
|
||
|
|
psutil
|
||
|
|
python-dotenv
|
||
|
|
pyyaml
|
||
|
|
pulp
|
||
|
|
requests
|
||
|
|
rich
|
||
|
|
tabulate
|
||
|
|
typing-extensions
|
||
|
|
wheel
|
||
|
|
];
|
||
|
|
|
||
|
|
meta = {
|
||
|
|
description = "Run LLMs and AI on any Cloud";
|
||
|
|
longDescription = ''
|
||
|
|
SkyPilot is a framework for running LLMs, AI, and batch jobs on any
|
||
|
|
cloud, offering maximum cost savings, highest GPU availability, and
|
||
|
|
managed execution.
|
||
|
|
'';
|
||
|
|
homepage = "https://github.com/skypilot-org/skypilot";
|
||
|
|
license = lib.licenses.asl20;
|
||
|
|
maintainers = with lib.maintainers; [ seanrmurphy ];
|
||
|
|
mainProgram = "sky";
|
||
|
|
};
|
||
|
|
}
|