diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 35b56f26a9..0539e255ea 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -371,16 +371,16 @@ 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._dynamo.config as config -for _try_dynamo_argument in torch_dynamo_arguments: - try: exec(_try_dynamo_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) +# except: pass +# pass torch_compile_options = { "epilogue_fusion" : True, "max_autotune" : True,