From 26614057a703a48d882f9c6c3956d613edb8387e Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 25 Jul 2024 11:21:17 +0200
Subject: [PATCH] Using g6 instead of g5. (#2281)

* Using g6 instead of g5.

* Update the idefics2 snapshot.
---
 .github/workflows/build.yaml                                | 2 +-
 .../test_idefics2/test_flash_idefics2_next_all_params.json  | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 89e3f18c..c289f63f 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -49,7 +49,7 @@ jobs:
                 export dockerfile="Dockerfile"
                 export label_extension=""
                 export docker_devices=""
-                export runs_on="aws-g5-12xlarge-plus"
+                export runs_on="aws-g6-12xlarge-plus"
                 ;;
             rocm)
                 export dockerfile="Dockerfile_amd"
diff --git a/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json b/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
index 45601505..dab437b9 100644
--- a/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
+++ b/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
@@ -30,7 +30,7 @@
       },
       {
         "id": 264,
-        "logprob": -0.37573242,
+        "logprob": -0.38061523,
         "special": false,
         "text": " a"
       },
@@ -42,7 +42,7 @@
       },
       {
         "id": 4480,
-        "logprob": -0.3322754,
+        "logprob": -0.26782227,
         "special": false,
         "text": " feature"
       },
@@ -78,7 +78,7 @@
       },
       {
         "id": 13,
-        "logprob": 0.0,
+        "logprob": -0.10632324,
         "special": false,
         "text": "\n"
       }