48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
"""
|
|
Unified core module for Unsloth backend
|
|
"""
|
|
|
|
# Inference
|
|
from .inference import InferenceBackend, get_inference_backend
|
|
|
|
# Training
|
|
from .training import UnslothTrainer, get_trainer, TrainingBackend, get_training_backend, create_training_handlers, TrainingProgress
|
|
|
|
# Configuration (from utils)
|
|
from utils.models import is_vision_model, ModelConfig, scan_trained_loras, load_model_defaults, get_base_model_from_lora
|
|
|
|
# Utilities (from utils)
|
|
from utils.paths import normalize_path, is_local_path, is_model_cached
|
|
from utils.utils import without_hf_auth, format_error_message, get_gpu_memory_info, search_hf_models
|
|
from utils.datasets.dataset_utils import format_and_template_dataset
|
|
|
|
__all__ = [
|
|
# Inference
|
|
'InferenceBackend',
|
|
'get_inference_backend',
|
|
|
|
# Training
|
|
'UnslothTrainer',
|
|
'get_trainer',
|
|
'get_training_backend',
|
|
'TrainingBackend',
|
|
'create_training_handlers',
|
|
'TrainingProgress',
|
|
|
|
# Config
|
|
'ModelConfig',
|
|
'is_vision_model',
|
|
'scan_trained_loras',
|
|
'load_model_defaults',
|
|
'get_base_model_from_lora',
|
|
|
|
# Utils
|
|
'search_hf_models',
|
|
'format_and_template_dataset',
|
|
'normalize_path',
|
|
'is_local_path',
|
|
'is_model_cached',
|
|
'without_hf_auth',
|
|
'format_error_message',
|
|
'get_gpu_memory_info',
|
|
]
|