p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #2474 from younesbelkada/add-hf-tags 

FEAT: add HF tags for models that have been trained with llama-factory
This commit is contained in:
hoshi-hiyouga 2024-02-14 10:26:03 +08:00 committed by GitHub
parent 12b2066e34 0ca0f08162
commit 8a1b389086
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/llmtuner/model/loader.py
View File

@ -87,6 +87,17 @@ def load_model_and_tokenizer(
**config_kwargs, **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) patch_model(model, tokenizer, model_args, is_trainable)
register_autoclass(config, model, tokenizer) register_autoclass(config, model, tokenizer)