Upload models/__init__.py with huggingface_hub

models/__init__.py

@@ -0,0 +1,23 @@
+"""
+Neural network models for Palette-Reasoner
+"""
+
+from .scope_pooler import ScopePooler, RegionMetadata
+from .trm_core import TRMCore, ReasoningTrace, StepOutput, MultiHeadAttention
+from .vit import VisionTransformer, PaletteFeatureExtractor, TransformerBlock
+from .edit_ops import PaletteEditOps, EditAction, OpCode
+from .hsl_feature_extractor import HSLFeatureExtractor
+from .hueai_semantic_encoder import HueAISemanticEncoder
+from .code_context_extractor import CodeContextExtractor
+from .hybrid_pooler import HybridRegionPooler
+from .fast_pooler import FastScopePooler
+from .oklab_utils import OKLabMSELoss
+
+__all__ = [
+    'ScopePooler', 'RegionMetadata',
+    'TRMCore', 'ReasoningTrace', 'StepOutput', 'MultiHeadAttention',
+    'VisionTransformer', 'PaletteFeatureExtractor', 'TransformerBlock',
+    'PaletteEditOps', 'EditAction', 'OpCode',
+    'HSLFeatureExtractor', 'HueAISemanticEncoder', 'CodeContextExtractor',
+    'HybridRegionPooler', 'FastScopePooler', 'OKLabMSELoss',
+]