Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "a2908d88d093a05bcdebe9011ba69b6dd48156ba"
Unverified Commit 86bcbc38 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Tests] increase transformers version in `test_low_cpu_mem_usage_with_loading` (#9662) 

increase transformers version in test_low_cpu_mem_usage_with_loading
parent 6a5f0648
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/lora/utils.py
View file @ 86bcbc38
...@@ -341,7 +341,7 @@ class PeftLoraLoaderMixinTests: ...@@ -341,7 +341,7 @@ class PeftLoraLoaderMixinTests:
self.assertTrue(output_lora.shape == self.output_shape) self.assertTrue(output_lora.shape == self.output_shape)
@require_peft_version_greater("0.13.1") @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): def test_low_cpu_mem_usage_with_loading(self):
"""Tests if we can load LoRA state dict with low_cpu_mem_usage.""" """Tests if we can load LoRA state dict with low_cpu_mem_usage."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment