Skip to content

GitLab

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

FIX [`bnb` / `tests`] Propagate the changes from #29092 to 4-bit tests (#29122) 

* forgot to push the changes for 4bit ..

* trigger CI
parent 1c9134f0
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/quantization/bnb/test_4bit.py
View file @ ff76e7c2
...@@ -43,7 +43,7 @@ from transformers.testing_utils import ( ...@@ -43,7 +43,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
elif model.config.model_type == "opt": elif model.config.model_type == "opt":
try: try:
......
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