From ff76e7c2126ab26e5722f54640d44cab7e3dfdd4 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Tue, 20 Feb 2024 11:11:15 +0100
Subject: [PATCH] FIX [`bnb` / `tests`] Propagate the changes from #29092 to
 4-bit tests (#29122)

* forgot to push the changes for 4bit ..

* trigger CI
---
 tests/quantization/bnb/test_4bit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/quantization/bnb/test_4bit.py b/tests/quantization/bnb/test_4bit.py
index 782e9a082fd7df..39d5598e37576a 100644
--- a/tests/quantization/bnb/test_4bit.py
+++ b/tests/quantization/bnb/test_4bit.py
@@ -43,7 +43,7 @@
 
 
 def get_some_linear_layer(model):
-    if model.config.model_type == "openai-community/gpt2":
+    if model.config.model_type == "gpt2":
         return model.transformer.h[0].mlp.c_fc
     elif model.config.model_type == "opt":
         try: