Skip to content
Merged
Changes from 1 commit
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
f6dfa80
add model registry
jeromeku Mar 28, 2025
a5e7b3a
move hf hub utils to unsloth/utils
jeromeku Mar 28, 2025
dc8f34e
refactor global model info dicts to dataclasses
jeromeku Mar 30, 2025
7cd2763
fix dataclass init
jeromeku Mar 30, 2025
9899a72
fix llama registration
jeromeku Mar 30, 2025
310c598
remove deprecated key function
jeromeku Mar 30, 2025
e70d035
start registry reog
jeromeku Mar 30, 2025
de1fe25
add llama vision
jeromeku Mar 30, 2025
7e2207c
quant types -> Enum
jeromeku Mar 30, 2025
c3a1aff
remap literal quant types to QuantType Enum
jeromeku Mar 30, 2025
03de6df
add llama model registration
jeromeku Mar 30, 2025
fa95aa0
fix quant tag mapping
jeromeku Mar 30, 2025
fdafa78
add qwen2.5 models to registry
jeromeku Mar 31, 2025
6049310
add option to include original model in registry
jeromeku Mar 31, 2025
8dc3d66
handle quant types per model size
jeromeku Mar 31, 2025
1237075
separate registration of base and instruct llama3.2
jeromeku Mar 31, 2025
baab018
add QwenQVQ to registry
jeromeku Mar 31, 2025
6b08fc3
add gemma3 to registry
jeromeku Mar 31, 2025
44e227b
add phi
jeromeku Mar 31, 2025
d633179
add deepseek v3
jeromeku Mar 31, 2025
0755b45
add deepseek r1 base
jeromeku Mar 31, 2025
17358e6
add deepseek r1 zero
jeromeku Mar 31, 2025
975d263
add deepseek distill llama
jeromeku Mar 31, 2025
229ae10
add deepseek distill models
jeromeku Mar 31, 2025
6439e88
remove redundant code when constructing model names
jeromeku Mar 31, 2025
4e1df71
add mistral small to registry
jeromeku Mar 31, 2025
6d4ede4
rename model registration methods
jeromeku Apr 1, 2025
a774726
rename deepseek registration methods
jeromeku Apr 1, 2025
a2a4366
refactor naming for mistral and phi
jeromeku Apr 1, 2025
02fbb87
add global register models
jeromeku Apr 1, 2025
7fbde42
refactor model registration tests for new registry apis
jeromeku Apr 1, 2025
a2d3ad9
add model search method
jeromeku Apr 1, 2025
13a1126
remove deprecated registration api
jeromeku Apr 1, 2025
4840a32
add quant type test
jeromeku Apr 1, 2025
7d64639
add registry readme
jeromeku Apr 1, 2025
12b0d32
make llama registration more specific
jeromeku Apr 1, 2025
ea75001
clear registry when executing individual model registration file
jeromeku Apr 1, 2025
d854070
more registry readme updates
jeromeku Apr 1, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove deprecated key function
  • Loading branch information
jeromeku committed Mar 30, 2025
commit 310c59800a1fb848893236f0a0aa55b01f77dcd5
14 changes: 0 additions & 14 deletions unsloth/model_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,20 +19,6 @@
_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
Expand Down