FIX [bnb / tests] Propagate the changes from #29092 to 4-bit tests (#29122)

* forgot to push the changes for 4bit .. * trigger CI
2024-02-20 11:11:15 +01:00
parent 1c9134f004
commit ff76e7c212
1 changed files with 1 additions and 1 deletions
--- a/tests/quantization/bnb/test_4bit.py
+++ b/tests/quantization/bnb/test_4bit.py
@@ -43,7 +43,7 @@ from transformers.testing_utils import (


 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: