From 303d32cfb5b1fa01b2a879304144b7213d8c8da5 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Oct 2024 14:01:49 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index a13723c9fb..0539e255ea 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -371,11 +371,11 @@ torch_dynamo_arguments = [ "config.cache_size_limit = 1024", # Flex Attention "config.inline_inbuilt_nn_modules = True", # Torch 2.5 Regional recompilation ] -import torch._inductor.config as config -for _try_compile_argument in torch_compile_arguments: - try: exec(_try_compile_argument) - except: pass -pass +# import torch._inductor.config as config +# for _try_compile_argument in torch_compile_arguments: +# try: exec(_try_compile_argument) +# except: pass +# pass # import torch._dynamo.config as config # for _try_dynamo_argument in torch_dynamo_arguments: # try: exec(_try_dynamo_argument)