File size: 601 Bytes
bfc6d2a | 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 | """Model loading and inference for A1-Max MuQ LoRA."""
from models.loader import (
A1MaxInferenceHead,
ModelCache,
get_model_cache,
)
from models.inference import (
extract_muq_embeddings,
predict_with_ensemble,
)
from models.calibration import (
calibrate_predictions,
predictions_to_calibrated_dict,
get_calibration_context,
)
__all__ = [
"A1MaxInferenceHead",
"ModelCache",
"get_model_cache",
"extract_muq_embeddings",
"predict_with_ensemble",
"calibrate_predictions",
"predictions_to_calibrated_dict",
"get_calibration_context",
]
|