1 files changed, 1 insertions, 1 deletions

diff --git a/train_lora.py b/train_lora.py
index 70fbae4..737af58 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -46,7 +46,7 @@ torch.backends.cudnn.benchmark = True
 torch._dynamo.config.log_level = logging.WARNING
 
 hidet.torch.dynamo_config.use_tensor_core(True)
-# hidet.torch.dynamo_config.use_attention(True)
+hidet.torch.dynamo_config.use_attention(True)
 hidet.torch.dynamo_config.search_space(0)