Skip to content

Commit

Permalink
FIX [bnb / tests] Propagate the changes from #29092 to 4-bit tests (
Browse files Browse the repository at this point in the history
#29122)

* forgot to push the changes for 4bit ..

* trigger CI
  • Loading branch information
younesbelkada authored Feb 20, 2024
1 parent 1c9134f commit ff76e7c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/quantization/bnb/test_4bit.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,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
elif model.config.model_type == "opt":
try:
Expand Down

0 comments on commit ff76e7c

Please sign in to comment.