From f0a171923cc8240177302f3dccb6177a2c646ab3 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 11 May 2023 18:37:43 +0200 Subject: Update --- train_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_lora.py') 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) -- cgit v1.2.3-54-g00ecf