Skip to content

GitLab

Unverified Commit a75a6c93 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

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

Update test_mixed_int8.py
parent 864c8e6e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/quantization/bnb/test_mixed_int8.py
View file @ a75a6c93
...@@ -42,7 +42,7 @@ from transformers.testing_utils import ( ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment