SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

FIX [bnb / tests]: Fix currently failing bnb tests (#29092)

Browse Source 
Update test_mixed_int8.py
This commit is contained in:
Younes Belkada
2024-02-19 10:39:12 +01:00
committed by GitHub
parent 864c8e6ea3
commit a75a6c9315
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/quantization/bnb/test_mixed_int8.py
View File

@@ -42,7 +42,7 @@ from transformers.testing_utils import (
def get_some_linear_layer(model): 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.c_fc
return model.transformer.h[0].mlp.dense_4h_to_h return model.transformer.h[0].mlp.dense_4h_to_h