Supports multiple base models and model flags