From c68a3bc9ecea534fa296a06ddf76ac00dace163d Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Wed, 31 Jan 2024 03:28:30 +1100 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 07396313b3..a9e5a2ca5c 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -71,7 +71,7 @@ class FastLanguageModel(FastLlamaModel): device_map = "sequential", rope_scaling = None, fix_tokenizer = True, - use_gradient_checkpointing = True, + use_gradient_checkpointing = False, *args, **kwargs, ): old_model_name = model_name