From a5b81ab457edc8378029df5bd26d6197475ec5ff Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sun, 15 Sep 2024 00:24:29 +0200
Subject: [PATCH] Home.

---
 .github/workflows/nix_integration_tests.yaml           |  1 +
 .../test_mamba/test_mamba_all_params.json              | 10 +++++-----
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml
index 239f5675..ff929f9c 100644
--- a/.github/workflows/nix_integration_tests.yaml
+++ b/.github/workflows/nix_integration_tests.yaml
@@ -39,3 +39,4 @@ jobs:
         nix develop .#test --command pytest -svv integration-tests/ --release
       env:
         HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        HF_HOME: /mnt/cache
diff --git a/integration-tests/models/__snapshots__/test_mamba/test_mamba_all_params.json b/integration-tests/models/__snapshots__/test_mamba/test_mamba_all_params.json
index ef88926c..5ffeb559 100644
--- a/integration-tests/models/__snapshots__/test_mamba/test_mamba_all_params.json
+++ b/integration-tests/models/__snapshots__/test_mamba/test_mamba_all_params.json
@@ -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"
       }