From 1c1461ae0990c2594fe33bfcb196ee33fabc7d41 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 10 Mar 2024 04:08:02 +1100 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 3d77d3103d..c1774eba6d 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -79,12 +79,14 @@ class FastLanguageModel(FastLlamaModel): ): old_model_name = model_name model_name = _get_model_name(model_name, load_in_4bit) + print(model_name) # First check if it's a normal model via AutoConfig is_peft = False try: model_config = AutoConfig.from_pretrained(model_name, token = token) is_peft = False + print(model_config) except: try: # Most likely a PEFT model @@ -96,6 +98,7 @@ class FastLanguageModel(FastLlamaModel): model_name = _get_model_name(peft_config.base_model_name_or_path, load_in_4bit) model_config = AutoConfig.from_pretrained(model_name, token = token) is_peft = True + print(model_config) pass model_type = model_config.model_type