ai_exec / src /inference /__init__.py
Chaitanya-aitf's picture
Upload 38 files
45ee481 verified
"""
Inference module for AI Executive System.
This module handles:
- Voice model loading and inference
- Refinement model loading and inference
- Dual-LLM pipeline orchestration
- Prompt templates management
"""
from .voice_model import VoiceModel
from .refinement_model import RefinementModel
from .dual_llm_pipeline import DualLLMPipeline
from .prompt_templates import (
VOICE_MODEL_SYSTEM_PROMPT,
REFINEMENT_MODEL_SYSTEM_PROMPT,
)
__all__ = [
"VoiceModel",
"RefinementModel",
"DualLLMPipeline",
"VOICE_MODEL_SYSTEM_PROMPT",
"REFINEMENT_MODEL_SYSTEM_PROMPT",
]