Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.6 kB · 99 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100{ lib, buildPythonPackage, fetchFromGitHub, setuptools, aiohttp, fastapi, httpx, markdown2, nh3, numpy, prompt-toolkit, pydantic, requests, rich, shortuuid, tiktoken, uvicorn, anthropic, openai, ray, wandb, einops, gradio, accelerate, peft, sentencepiece, torch, transformers, protobuf,}:let version = "0.2.36";inbuildPythonPackage { pname = "fschat"; inherit version; format = "pyproject";
src = fetchFromGitHub { owner = "lm-sys"; repo = "FastChat"; tag = "v${version}"; hash = "sha256-tQuvQXzQbQjU16DfS1o55VHW6eklngEvIigzZGgrKB8="; };
nativeBuildInputs = [ setuptools ];
propagatedBuildInputs = [ aiohttp fastapi httpx markdown2 nh3 numpy prompt-toolkit pydantic requests rich shortuuid tiktoken uvicorn # ] ++ markdown2.optional-dependencies.all; ];
optional-dependencies = { llm_judge = [ anthropic openai ray ]; train = [ # flash-attn wandb einops ]; webui = [ gradio ]; model_worker = [ accelerate peft sentencepiece torch transformers protobuf ]; };
pythonImportsCheck = [ "fastchat" ];
# tests require networking doCheck = false;
meta = with lib; { description = "Open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena"; homepage = "https://github.com/lm-sys/FastChat"; license = licenses.asl20; maintainers = with maintainers; [ happysalada ]; };}