SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

chore: fix typos in tests directory (#36785)

Browse Source 
* chore: fix typos in tests directory

* chore: fix typos in tests directory

* chore: fix typos in tests directory

* chore: fix typos in tests directory

* chore: fix typos in tests directory

* chore: fix typos in tests directory

* chore: fix typos in tests directory
This commit is contained in:
Afanti
2025-03-18 17:31:13 +08:00
committed by GitHub
parent 7f5077e536
commit 19b9d8ae13
12 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -179,7 +179,7 @@ class Bnb4BitTest(Base4bitTest):
def test_original_dtype(self):
r"""
A simple test to check if the model succesfully stores the original dtype
A simple test to check if the model successfully stores the original dtype
"""
self.assertTrue(hasattr(self.model_4bit.config, "_pre_quantization_dtype"))
self.assertFalse(hasattr(self.model_fp16.config, "_pre_quantization_dtype"))
@@ -496,8 +496,8 @@ class Pipeline4BitTest(Base4bitTest):
def test_pipeline(self):
r"""
The aim of this test is to verify that the mixed 4bit is compatible with `pipeline` from transformers. Since
we used pipline for inference speed benchmarking we want to make sure that this feature does not break anything
on pipline.
we used pipeline for inference speed benchmarking we want to make sure that this feature does not break anything
on pipeline.
"""
# self._clear_cuda_cache()
self.pipe = pipeline(