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/README.md
View File

@@ -13,7 +13,7 @@ The following is the recipe on how to effectively debug `bitsandbytes` integrati
The following instructions are tested with 2 NVIDIA-Tesla T4 GPUs. To run successfully `bitsandbytes` you would need a 8-bit core tensor supported GPU. Note that Turing, Ampere or newer architectures - e.g. T4, RTX20s RTX30s, A40-A100, A6000 should be supported.
## Virutal envs
## Virtual envs
```bash
conda create --name int8-testing python==3.8
@@ -61,7 +61,7 @@ This happens when some Linear weights are set to the CPU when using `accelerate`
Use the latest version of `accelerate` with a command such as: `pip install -U accelerate` and the problem should be solved.
### `Parameter has no attribue .CB`
### `Parameter has no attribute .CB`
Same solution as above.
@@ -71,7 +71,7 @@ Run your script by pre-pending `CUDA_LAUNCH_BLOCKING=1` and you should observe a
### `CUDA illegal memory error: an illegal memory access at line...`:
Check the CUDA verisons with:
Check the CUDA versions with:
```bash
nvcc --version
```

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(

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

@@ -213,7 +213,7 @@ class MixedInt8Test(BaseMixedInt8Test):
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_8bit.config, "_pre_quantization_dtype"))
self.assertFalse(hasattr(self.model_fp16.config, "_pre_quantization_dtype"))
@@ -655,8 +655,8 @@ class MixedInt8TestPipeline(BaseMixedInt8Test):
def test_pipeline(self):
r"""
The aim of this test is to verify that the mixed int8 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(