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

Attempt.

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-16 11:16:03 +02:00
parent ec6fe324c6
commit 332e42f59a
No known key found for this signature in database
GPG Key ID: 64AF4752B2967863
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/nix_integration_tests.yaml vendored
View File

@ -33,7 +33,7 @@ jobs:
- name: Python tests. - name: Python tests.
run: | run: |
sudo mkdir -p /run/opengl-driver/lib/ sudo mkdir -p /run/opengl-driver/lib/
export NVIDIA=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11') export NVIDIA=$(nix build --print-out-paths .#nvidia-driver)
sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \; sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \; sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
nix develop .#test --command pytest -svv integration-tests/ --release nix develop .#test --command pytest -svv integration-tests/ --release

1
flake.nix
View File

@ -149,6 +149,7 @@
}; };
}; };
packages.nvidia-driver = pkgs.linuxPackages.nvidia_x11;
packages.default = pkgs.writeShellApplication { packages.default = pkgs.writeShellApplication {
name = "text-generation-inference"; name = "text-generation-inference";
runtimeInputs = [ runtimeInputs = [