| """
|
| COGNITIVE-CORE Framework
|
| ========================
|
|
|
| Universal template for Ame Web Studio's cognitive AI architectures.
|
| Provides standardized loading, checkpoint management, and utilities
|
| for vision, language, world model, and multimodal cognitive systems.
|
|
|
| Copyright © 2026 Mike Amega (Logo) - Ame Web Studio
|
| License: Proprietary - All Rights Reserved
|
| """
|
|
|
| from .cognitive_base import (
|
| CognitiveConfig,
|
| CognitiveModule,
|
| MemoryModule,
|
| TemporalModule,
|
| WorldModelModule,
|
| CognitivePreTrainedModel,
|
| register_cognitive_model,
|
| )
|
|
|
| from .cognitive_checkpoint import (
|
| remap_checkpoint_keys,
|
| validate_checkpoint,
|
| save_cognitive_checkpoint,
|
| load_cognitive_checkpoint,
|
| )
|
|
|
| from .cognitive_utils import (
|
| setup_environment,
|
| get_device,
|
| get_optimal_dtype,
|
| get_memory_info,
|
| clear_memory,
|
| estimate_model_memory,
|
| print_model_info,
|
| print_training_progress,
|
| get_hf_token,
|
| )
|
|
|
| from .cognitive_training import (
|
| CognitiveTrainingConfig,
|
| CognitiveTrainer,
|
| prepare_dataset,
|
| create_instruction_dataset,
|
| quick_train,
|
| CognitiveStateCallback,
|
| )
|
|
|
| __version__ = "1.0.0"
|
| __author__ = "Mike Amega"
|
| __license__ = "Proprietary"
|
|
|
| __all__ = [
|
|
|
| "CognitiveConfig",
|
| "CognitiveModule",
|
| "MemoryModule",
|
| "TemporalModule",
|
| "WorldModelModule",
|
| "CognitivePreTrainedModel",
|
| "register_cognitive_model",
|
|
|
| "remap_checkpoint_keys",
|
| "validate_checkpoint",
|
| "save_cognitive_checkpoint",
|
| "load_cognitive_checkpoint",
|
|
|
| "setup_environment",
|
| "get_device",
|
| "get_optimal_dtype",
|
| "get_memory_info",
|
| "clear_memory",
|
| "estimate_model_memory",
|
| "print_model_info",
|
| "print_training_progress",
|
| "get_hf_token",
|
|
|
| "CognitiveTrainingConfig",
|
| "CognitiveTrainer",
|
| "prepare_dataset",
|
| "create_instruction_dataset",
|
| "quick_train",
|
| "CognitiveStateCallback",
|
| ]
|
|
|
