Update loader.py

2024-11-26 00:18:00 -08:00 · 2024-11-26 00:18:00 -08:00 · 77fc0c96c3
commit 77fc0c96c3
parent 34ea5bc89f
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@ -465,6 +465,7 @@ class FastVisionModel(FastBaseVisionModel):
        if is_peft:
            # Check base model again for PEFT
            model_name = get_model_name(peft_config.base_model_name_or_path, load_in_4bit)
+            print(model_name)
            model_config = AutoConfig.from_pretrained(
                model_name,
                token = token,