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

Home.

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-15 00:24:29 +02:00
parent 98f2241a88
commit a5b81ab457
No known key found for this signature in database
GPG Key ID: 64AF4752B2967863
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -39,3 +39,4 @@ jobs:
nix develop .#test --command pytest -svv integration-tests/ --release
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
HF_HOME: /mnt/cache

10
integration-tests/models/__snapshots__/test_mamba/test_mamba_all_params.json
View File

@ -52,7 +52,7 @@
},
{
"id": 9830,
"logprob": -2.25,
"logprob": -2.359375,
"special": false,
"text": " colors"
},
@ -64,13 +64,13 @@
},
{
"id": 329,
"logprob": -2.171875,
"logprob": -2.296875,
"special": false,
"text": " A"
},
{
"id": 1180,
"logprob": -2.046875,
"logprob": -1.84375,
"special": false,
"text": " number"
},
@ -82,13 +82,13 @@
},
{
"id": 1027,
"logprob": -1.5546875,
"logprob": -1.1015625,
"special": false,
"text": " different"
},
{
"id": 3295,
"logprob": -0.97265625,
"logprob": -1.0546875,
"special": false,
"text": " color"
}