huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 00:12:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Attempt at integration tests.

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-12 14:47:18 +02:00
parent f47cdc1fe1
commit 3d73c99ebe
No known key found for this signature in database
GPG Key ID: 64AF4752B2967863
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flake.nix
View File

@ -67,6 +67,7 @@
'';
};
server = pkgs.python3.pkgs.callPackage ./nix/server.nix { inherit nix-filter; };
client = pkgs.python3.pkgs.callPackage ./nix/client.nix { inherit nix-filter; };
in
{
formatter = pkgs.nixfmt-rfc-style;
@ -88,6 +89,7 @@
launcher
router
server
client
openssl.dev
pkg-config
cargo

26
nix/client.nix Normal file
View File

@ -0,0 +1,26 @@
{
buildPythonPackage,
poetry-core,
huggingface-hub,
pydantic,
}:
buildPythonPackage {
name = "text-generation";
src = ../clients/python;
pyproject = true;
build-system = [ poetry-core ];
nativeBuildInputs = [ ];
pythonRemoveDeps = [ ];
dependencies = [
huggingface-hub
transformers
pydantic
];
}