We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
test_low_cpu_mem_usage_with_loading
1 parent 6a5f064 commit 86bcbc3Copy full SHA for 86bcbc3
tests/lora/utils.py
@@ -341,7 +341,7 @@ def test_low_cpu_mem_usage_with_injection(self):
341
self.assertTrue(output_lora.shape == self.output_shape)
342
343
@require_peft_version_greater("0.13.1")
344
- @require_transformers_version_greater("4.45.1")
+ @require_transformers_version_greater("4.45.2")
345
def test_low_cpu_mem_usage_with_loading(self):
346
"""Tests if we can load LoRA state dict with low_cpu_mem_usage."""
347
0 commit comments