move hf hub utils to unsloth/utils
This commit is contained in:
parent
6f6a7a5e9b
commit
f21d61ae5d
4 changed files with 5 additions and 2 deletions
|
|
@ -32,6 +32,10 @@ include-package-data = false
|
|||
exclude = ["images*", "tests*"]
|
||||
|
||||
[project.optional-dependencies]
|
||||
dev = [
|
||||
"pytest",
|
||||
]
|
||||
|
||||
triton = [
|
||||
"triton-windows ; platform_system == 'Windows'",
|
||||
]
|
||||
|
|
|
|||
|
|
@ -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",
|
||||
|
|
|
|||
0
unsloth/utils/__init__.py
Normal file
0
unsloth/utils/__init__.py
Normal file
Loading…
Add table
Add a link
Reference in a new issue