mirror of https://github.com/razor-ai/soup.git
43 lines
960 B
Python
43 lines
960 B
Python
"""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,
|
|
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",
|
|
"parse_gpu_budget",
|
|
"write_yaml",
|
|
]
|