Enable HPUGraphs for lora long-contexts tests

HabanaAI · Nov 4, 2024 · 0cc72b9 · 0cc72b9
1 parent 6643aa6
commit 0cc72b9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/lora/test_long_context_hpu.py b/tests/lora/test_long_context_hpu.py
@@ -116,7 +116,6 @@ def lora_llm(long_context_infos):
         long_lora_scaling_factors=tuple(scaling_factors),
         max_num_batched_tokens=4096 * 8,
         tensor_parallel_size=1,
-        enforce_eager=True,  # TODO Remove after SW-205153 is fixed
         dtype="bfloat16",
         disable_async_output_proc=True,  # TODO Remove after SW-204469 is fixed.
         distributed_executor_backend="mp")