From 0ca0f08162b18d326787939f12bb5ba07904fb4a Mon Sep 17 00:00:00 2001
From: younesbelkada <younesbelkada@gmail.com>
Date: Tue, 13 Feb 2024 05:58:49 +0000
Subject: [PATCH] add v1 hf tags

---
 src/llmtuner/model/loader.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/llmtuner/model/loader.py b/src/llmtuner/model/loader.py
index deeaed40..6d75c15b 100644
--- a/src/llmtuner/model/loader.py
+++ b/src/llmtuner/model/loader.py
@@ -87,6 +87,17 @@ def load_model_and_tokenizer(
             **config_kwargs,
         )
 
+        # Add llama-factory tag to push these tags on the Hub.
+        # the feature is available since 4.37.0 but adding the check
+        # just in case
+        if hasattr(model, "add_model_tags"):
+            model.add_model_tags(["llama-factory"])
+        else:
+            logger.warning_once(
+                "Was not able to properly tag the model, if you want to use the model tagging feature, make sure to "
+                "have transformers>=4.37.0 installed on your environment."
+            )
+
     patch_model(model, tokenizer, model_args, is_trainable)
     register_autoclass(config, model, tokenizer)