From bbdbcc9b8fae51a87e67c8b20afd7deb513f9b10 Mon Sep 17 00:00:00 2001 From: GM Harshvardhan Date: Thu, 29 Feb 2024 09:55:12 -0500 Subject: [PATCH] edit docstring test_lora_bnb_4bit_quantization --- tests/test_common_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_common_gpu.py b/tests/test_common_gpu.py index 0fbcad0c73..cf15bc0447 100644 --- a/tests/test_common_gpu.py +++ b/tests/test_common_gpu.py @@ -321,7 +321,7 @@ def test_lora_gptq_quantization_from_pretrained_safetensors(self): @pytest.mark.single_gpu_tests def test_lora_bnb_4bit_quantization(self): r""" - Test that tests if the 4bit quantization for Linear and Embedding using LoRA works as expected + Tests if the 4bit quantization for Linear and Embedding using LoRA works as expected """ whisper_4bit = WhisperForConditionalGeneration.from_pretrained( self.audio_model_id,