Skip to content

Commit 83701de

Browse files
committed
explicitly specify low_cpu_mem_usage=False.
1 parent efa33e3 commit 83701de

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/lora/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,7 @@ def test_low_cpu_mem_usage_with_loading(self):
385385

386386
self.assertTrue(os.path.isfile(os.path.join(tmpdirname, "pytorch_lora_weights.bin")))
387387
pipe.unload_lora_weights()
388-
pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.bin"))
388+
pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.bin"), low_cpu_mem_usage=False)
389389

390390
for module_name, module in modules_to_save.items():
391391
self.assertTrue(check_if_lora_correctly_set(module), f"Lora not correctly set in {module_name}")

0 commit comments

Comments
 (0)