src/models/__init__.py

# Export model instances and loading functions
try:
    from .model_loader import (
        # Model instances
        RTDETR_MODEL,
        RTDETR_PROCESSOR,
        HEAD_MODEL,
        HEAD_PROCESSOR,
        RMBG_MODEL,
        YOLOS_MODEL,
        YOLOS_PROCESSOR,
        
        # State variables
        MODELS_LOADED,
        LOAD_ERROR,
        LOAD_LOCK,
        DEVICE,
        
        # Functions
        load_models,
        ensure_models_loaded,
        check_hardware_environment,
        
        # Constants
        RTDETR_REPO,
        HEAD_DETECTION_REPO,
        RMBG_REPO,
        YOLOS_FASHIONPEDIA_REPO,
        
        # Error messages
        ERROR_NO_HF_TOKEN,
        ERROR_ACCESS_DENIED,
        ERROR_AUTH_FAILED,
        
        # Model configuration constants
        RTDETR_FULL_PRECISION,
        HEAD_DETECTION_FULL_PRECISION,
        RMBG_FULL_PRECISION,
        YOLOS_FASHIONPEDIA_FULL_PRECISION,
        
        # Optimization settings
        USE_TORCH_COMPILE,
        TORCH_COMPILE_MODE,
        TORCH_COMPILE_BACKEND,
        ENABLE_CHANNELS_LAST,
        ENABLE_CUDA_GRAPHS,
        USE_MIXED_PRECISION
    )
except ImportError:
    RTDETR_MODEL = None
    RTDETR_PROCESSOR = None
    HEAD_MODEL = None
    HEAD_PROCESSOR = None
    RMBG_MODEL = None
    YOLOS_MODEL = None
    YOLOS_PROCESSOR = None
    MODELS_LOADED = False
    LOAD_ERROR = ""
    LOAD_LOCK = None
    DEVICE = "cpu"
    load_models = lambda: None
    ensure_models_loaded = lambda: None
    check_hardware_environment = lambda: None
    RTDETR_REPO = "PekingU/rtdetr_r50vd"
    HEAD_DETECTION_REPO = "sanali209/DT_face_head_char"
    RMBG_REPO = "briaai/RMBG-2.0"
    YOLOS_FASHIONPEDIA_REPO = "valentinafeve/yolos-fashionpedia"
    ERROR_NO_HF_TOKEN = "HF_TOKEN environment variable not set."
    ERROR_ACCESS_DENIED = "Access denied to RMBG-2.0."
    ERROR_AUTH_FAILED = "Authentication failed."
    
    RTDETR_FULL_PRECISION = True
    HEAD_DETECTION_FULL_PRECISION = True
    RMBG_FULL_PRECISION = True
    YOLOS_FASHIONPEDIA_FULL_PRECISION = True
    
    USE_TORCH_COMPILE = True
    TORCH_COMPILE_MODE = "reduce-overhead"
    TORCH_COMPILE_BACKEND = "inductor"
    ENABLE_CHANNELS_LAST = True
    ENABLE_CUDA_GRAPHS = True
    USE_MIXED_PRECISION = True

__all__ = [
    # Model instances
    'RTDETR_MODEL',
    'RTDETR_PROCESSOR',
    'HEAD_MODEL',
    'HEAD_PROCESSOR',
    'RMBG_MODEL',
    'YOLOS_MODEL',
    'YOLOS_PROCESSOR',
    
    # State variables
    'MODELS_LOADED',
    'LOAD_ERROR',
    'LOAD_LOCK',
    'DEVICE',
    
    # Functions
    'load_models',
    'ensure_models_loaded',
    'check_hardware_environment',
    
    # Constants
    'RTDETR_REPO',
    'HEAD_DETECTION_REPO',
    'RMBG_REPO',
    'YOLOS_FASHIONPEDIA_REPO',
    
    # Error messages
    'ERROR_NO_HF_TOKEN',
    'ERROR_ACCESS_DENIED',
    'ERROR_AUTH_FAILED',
    
    # Model configuration constants
    'RTDETR_FULL_PRECISION',
    'HEAD_DETECTION_FULL_PRECISION',
    'RMBG_FULL_PRECISION',
    'YOLOS_FASHIONPEDIA_FULL_PRECISION',
    
    # Optimization settings
    'USE_TORCH_COMPILE',
    'TORCH_COMPILE_MODE',
    'TORCH_COMPILE_BACKEND',
    'ENABLE_CHANNELS_LAST',
    'ENABLE_CUDA_GRAPHS',
    'USE_MIXED_PRECISION'
]