From ce19ea56be98bdd91221c4df6792e77020707409 Mon Sep 17 00:00:00 2001 From: jeromeku Date: Mon, 31 Mar 2025 17:11:35 -0700 Subject: [PATCH] add global register models --- unsloth/registry/__init__.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/unsloth/registry/__init__.py b/unsloth/registry/__init__.py index dd5b45c4ee..154cea6deb 100644 --- a/unsloth/registry/__init__.py +++ b/unsloth/registry/__init__.py @@ -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() \ No newline at end of file