riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

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

Browse Source 
* forgot to push the changes for 4bit ..

* trigger CI
This commit is contained in:
Younes Belkada 2024-02-20 11:11:15 +01:00 committed by GitHub
parent 1c9134f004
commit ff76e7c212
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -43,7 +43,7 @@ from transformers.testing_utils import (
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: