Elgato_dark/unsloth
1
0
Fork 0
mirror of https://github.com/unslothai/unsloth synced 2026-04-21 13:37:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add HF tagging in unsloth (#170)

Browse source
This commit is contained in:
Younes Belkada 2024-02-13 08:28:42 +01:00 committed by GitHub
parent 3d5cf373bc
commit 474fd32f91
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/loader.py
View file

@ -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 = \