From c6ad5f97a3128197cd9f8f33e3ab2611fe11a8ef Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 20:09:11 +1100 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index f11a34a1b1..f48646ab45 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -118,9 +118,13 @@ class FastLanguageModel(FastLlamaModel): *args, **kwargs, ) - # in case the model supports tagging, add the unsloth tag. + # In case the model supports tagging, add the unsloth tag. if hasattr(model, "add_model_tags"): - model.add_model_tags(["unsloth"]) + model.add_model_tags(["unsloth",]) + pass + if hasattr(tokenizer, "add_model_tags"): + tokenizer.add_model_tags(["unsloth",]) + pass if load_in_4bit: # Fix up bitsandbytes config