AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
increase transformers version in test_low_cpu_mem_usage_with_loading
This commit is contained in:
Sayak Paul
2024-10-13 22:39:38 +05:30
committed by GitHub
parent 6a5f06488c
commit 86bcbc389e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/lora/utils.py
View File

@@ -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."""