ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove deprecated key function

Browse source
This commit is contained in:
jeromeku 2025-03-30 11:06:59 -07:00
commit ab77cc7c0d
1 changed files with 0 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

14
unsloth/model_registry.py
View file

@ -19,20 +19,6 @@ _IS_PHI_REGISTERED = False
_IS_PHI_INSTRUCT_REGISTERED = False
def construct_model_key(
org, base_name, version, size, quant_type, instruct_tag
):
key = f"{org}/{base_name}-{version}-{size}B"
if instruct_tag:
key = "-".join([key, instruct_tag])
if quant_type:
if quant_type == "bnb":
key = "-".join([key, BNB_QUANTIZED_TAG])
elif quant_type == "unsloth":
key = "-".join([key, UNSLOTH_DYNAMIC_QUANT_TAG])
return key
@dataclass
class ModelInfo:
org: str