add HF tagging in unsloth (#170)
This commit is contained in:
parent
1191e83866
commit
077e9dc87d
1 changed files with 4 additions and 0 deletions
|
|
@ -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 = \
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue