From 077e9dc87db44279e3cda9754d5fddcae35661a5 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Tue, 13 Feb 2024 08:28:42 +0100
Subject: [PATCH] add HF tagging in unsloth (#170)

---
 unsloth/models/loader.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index 674a5efa6f..f11a34a1b1 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -118,6 +118,10 @@ class FastLanguageModel(FastLlamaModel):
             *args, **kwargs,
         )
 
+        # in case the model supports tagging, add the unsloth tag.
+        if hasattr(model, "add_model_tags"):
+            model.add_model_tags(["unsloth"])
+
         if load_in_4bit:
             # Fix up bitsandbytes config
             quantization_config = \