diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py index 5102e65ae..9999d742d 100644 --- a/tests/e2e/multigpu/test_llama.py +++ b/tests/e2e/multigpu/test_llama.py @@ -155,6 +155,7 @@ def test_dpo_lora_ddp(self, temp_dir): "sample_packing": False, "eval_sample_packing": False, "pad_to_sequence_len": True, + "load_in_8bit": True, "adapter": "lora", "lora_r": 8, "lora_alpha": 16,