From 32aee20e3cade7068c737b4ed817c4f0db91195f Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Tue, 20 Feb 2024 02:28:10 +0000 Subject: [PATCH 1/2] forgot to push the changes for 4bit .. --- 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: From b900a1ca01c206448115f576d13d87073a012936 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Tue, 20 Feb 2024 06:44:55 +0000 Subject: [PATCH 2/2] trigger CI