Skip to content

Commit 15a6d1f

Browse files
committed
update.
1 parent 8a98954 commit 15a6d1f

1 file changed

Lines changed: 35 additions & 7 deletions

File tree

src/hf_model_inspector/__init__.py

Lines changed: 35 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,43 @@
1-
from .hf_model_inspector import (
2-
get_model_stats,
3-
get_model_usage,
1+
# Core loader
2+
from .loader import HFModelLoader
3+
4+
# Analyzer functions
5+
from .analyzer import (
6+
estimate_param_count,
7+
detect_quant_and_precision,
8+
analyze_tokenizer,
9+
extract_architecture_extras,
10+
)
11+
12+
# Formatter functions
13+
from .formatter import format_markdown, save_outputs
14+
15+
# Main API functions
16+
from .main import (
417
get_model_report_json,
518
get_model_report_md,
6-
recommend_models,
19+
save_model_report,
20+
get_lora_info,
721
)
822

23+
# Public API
924
__all__ = [
10-
"get_model_stats",
11-
"get_model_usage",
25+
# Main API
1226
"get_model_report_json",
1327
"get_model_report_md",
14-
"recommend_models",
28+
"save_model_report",
29+
"get_lora_info",
30+
31+
# Core classes
32+
"HFModelLoader",
33+
34+
# Analyzer
35+
"estimate_param_count",
36+
"detect_quant_and_precision",
37+
"analyze_tokenizer",
38+
"extract_architecture_extras",
39+
40+
# Formatter
41+
"format_markdown",
42+
"save_outputs",
1543
]

0 commit comments

Comments
 (0)