From 10fddaf59a1b1a495838fa29592819af3eab7e74 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Mon, 19 Feb 2024 02:13:00 +0100 Subject: [PATCH] Update test_mixed_int8.py --- tests/quantization/bnb/test_mixed_int8.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py index b926c80398c25a..1c4362ef0cec1a 100644 --- a/tests/quantization/bnb/test_mixed_int8.py +++ b/tests/quantization/bnb/test_mixed_int8.py @@ -42,7 +42,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 return model.transformer.h[0].mlp.dense_4h_to_h