Soup/src/soup_cli/autopilot/__init__.py at main · ghzhost/Soup · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
"""Autopilot — zero-config fine-tuning decision engine (Part H of v0.25.0)."""

from soup_cli.autopilot.analyzer import (
    DatasetProfile,
    HardwareProfile,
    ModelProfile,
    analyze_dataset,
    analyze_hardware,
    analyze_model,
)
from soup_cli.autopilot.decisions import (
    decide_batch_size,
    decide_epochs,
    decide_lr,
    decide_max_length,
    decide_peft,
    decide_performance_flags,
    decide_quantization,
    decide_task,
    detect_prequantized_format,
    detect_prequantized_format_from_path,
    parse_gpu_budget,
)
from soup_cli.autopilot.generate_config import build_soup_config, write_yaml

__all__ = [
    "DatasetProfile",
    "HardwareProfile",
    "ModelProfile",
    "analyze_dataset",
    "analyze_hardware",
    "analyze_model",
    "build_soup_config",
    "decide_batch_size",
    "decide_epochs",
    "decide_lr",
    "decide_max_length",
    "decide_peft",
    "decide_performance_flags",
    "decide_quantization",
    "decide_task",
    "detect_prequantized_format",
    "detect_prequantized_format_from_path",
    "parse_gpu_budget",
    "write_yaml",
]