From 86bcbc389e810c3d12a61dce57dd02071c6b3050 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sun, 13 Oct 2024 22:39:38 +0530 Subject: [PATCH] [Tests] increase transformers version in `test_low_cpu_mem_usage_with_loading` (#9662) increase transformers version in test_low_cpu_mem_usage_with_loading --- tests/lora/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lora/utils.py b/tests/lora/utils.py index 9c982e8de3..48c7d5fcec 100644 --- a/tests/lora/utils.py +++ b/tests/lora/utils.py @@ -341,7 +341,7 @@ class PeftLoraLoaderMixinTests: self.assertTrue(output_lora.shape == self.output_shape) @require_peft_version_greater("0.13.1") - @require_transformers_version_greater("4.45.1") + @require_transformers_version_greater("4.45.2") def test_low_cpu_mem_usage_with_loading(self): """Tests if we can load LoRA state dict with low_cpu_mem_usage."""