move hf hub utils to unsloth/utils

This commit is contained in:
jeromeku 2025-03-28 16:54:38 -07:00
commit f21d61ae5d
4 changed files with 5 additions and 2 deletions

View file

@ -32,6 +32,10 @@ include-package-data = false
exclude = ["images*", "tests*"]
[project.optional-dependencies]
dev = [
"pytest",
]
triton = [
"triton-windows ; platform_system == 'Windows'",
]

View file

@ -11,8 +11,7 @@ from unsloth.model_registry import (
get_qwen_models,
get_qwen_vl_models,
)
from .utils.hf_hub import get_model_info
from unsloth.utils.hf_hub import get_model_info
MODEL_NAMES = [
"llama",

View file