diff --git a/.github/workflows/push-important-models.yml b/.github/workflows/push-important-models.yml
index 92cd3badd86957..83c7eb2450fc97 100644
--- a/.github/workflows/push-important-models.yml
+++ b/.github/workflows/push-important-models.yml
@@ -52,11 +52,12 @@ jobs:
             echo "matrix=[$matrix_string]" >> $GITHUB_OUTPUT
   test_modified_files:
     needs: get_modified_models
-    name: Slow & FA2 tests
-    runs-on: [single-gpu, nvidia-gpu, a10, ci]
-    container:
-      image: huggingface/transformers-all-latest-gpu
-      options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
+    name: Slow & FA2 tests ${{ matrix.model-name }}
+    runs-on: ubuntu-latest
+    # runs-on: [single-gpu, nvidia-gpu, a10, ci]
+    # container:
+      # image: huggingface/transformers-all-latest-gpu
+      # options: --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
     if: ${{ needs.get_modified_models.outputs.matrix != '[]' && needs.get_modified_models.outputs.matrix != '' }}
     strategy:
       fail-fast: false
@@ -66,7 +67,7 @@ jobs:
     steps:
       - name: Echo
         run: |
-          echo ${{ needs.get_modified_models.outputs.matrix }}
+          echo ${{ matrix.model-name }}
       - name: Check out code
         uses: actions/checkout@v3
       
diff --git a/src/transformers/models/bark/modeling_bark.py b/src/transformers/models/bark/modeling_bark.py
index de04614075cf80..c004b23e0bfc36 100644
--- a/src/transformers/models/bark/modeling_bark.py
+++ b/src/transformers/models/bark/modeling_bark.py
@@ -16,6 +16,7 @@
 import math
 from typing import Dict, Optional, Tuple, Union
 
+
 import numpy as np
 import torch
 from torch import nn