unsloth/studio/backend/models/__init__.py

27 lines
485 B
Python

"""
Pydantic models for API request/response schemas
"""
from .training import (
TrainingStartRequest,
TrainingJobResponse,
TrainingStatus,
TrainingProgress,
)
from .models import (
ModelDetails,
LoRAInfo,
LoRAScanResponse,
)
__all__ = [
# Training schemas
"TrainingStartRequest",
"TrainingJobResponse",
"TrainingStatus",
"TrainingProgress",
# Model management schemas
"ModelDetails",
"LoRAInfo",
"LoRAScanResponse",
]