"examples/vscode:/vscode.git/clone" did not exist on "0f58903bb62870342eae52f5a02c9105ec6f9b1e"
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
...@@ -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