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

add global register models

Browse source
This commit is contained in:
jeromeku 2025-03-31 17:11:35 -07:00
commit ce19ea56be
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
unsloth/registry/__init__.py
View file

@ -1,5 +1,13 @@
# from ._deepseek import register_deepseek_models, register
# from ._llama import register_llama_models, register_llama_vision_models
# from ._mistral import register_mistral_models
# from ._openai import register_openai_models
# from ._qwen import register_qwen_models
from ._deepseek import register_deepseek_models as _register_deepseek_models
from ._gemma import register_gemma_models as _register_gemma_models
from ._llama import register_llama_models as _register_llama_models
from ._mistral import register_mistral_models as _register_mistral_models
from ._phi import register_phi_models as _register_phi_models
from ._qwen import register_qwen_models as _register_qwen_models
_register_deepseek_models()
_register_gemma_models()
_register_llama_models()
_register_mistral_models()
_register_phi_models()
_register_qwen_models()