Upload configuration_nexus_worldmodel.py with huggingface_hub

configuration_nexus_worldmodel.py

@@ -0,0 +1,70 @@
+"""
+NEXUS-WorldModel Configuration
+
+Auto-generated configuration class for loading from HuggingFace Hub.
+"""
+
+import json
+import os
+from dataclasses import dataclass, field
+from typing import List, Dict, Any, Optional
+
+@dataclass
+class WorldConfig:
+    width: int = 64
+    height: int = 64
+    channels: int = 3
+    gravity: float = 0.1
+    friction: float = 0.98
+    bounce: float = 0.8
+    max_velocity: float = 5.0
+    max_agents: int = 5
+    max_obstacles: int = 10
+    max_zones: int = 3
+    agent_radius: float = 2.0
+    dt: float = 1.0
+
+@dataclass
+class NexusWorldModelConfig:
+    """Configuration for NEXUS-WorldModel v2.0"""
+    
+    model_type: str = "nexus-worldmodel"
+    version: str = "2.0"
+    
+    d_model: int = 512
+    d_ff: int = 2048
+    n_layers: int = 8
+    n_heads: int = 8
+    dropout: float = 0.1
+    
+    latent_dim: int = 256
+    latent_state_dim: int = 256
+    action_dim: int = 5
+    
+    use_lpol: bool = True
+    use_gqa: bool = True
+    gqa_num_heads: int = 8
+    gqa_num_kv_groups: int = 2
+    
+    neurogenesis_enabled: bool = True
+    dream_enabled: bool = True
+    
+    world: WorldConfig = field(default_factory=WorldConfig)
+    
+    def to_dict(self) -> Dict[str, Any]:
+        return {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
+    
+    @classmethod
+    def from_pretrained(cls, path_or_repo: str, **kwargs):
+        """Load config from HuggingFace Hub or local path"""
+        if os.path.isdir(path_or_repo):
+            config_file = os.path.join(path_or_repo, "config.json")
+        else:
+            from huggingface_hub import hf_hub_download
+            config_file = hf_hub_download(repo_id=path_or_repo, filename="config.json", **kwargs)
+        
+        with open(config_file, "r") as f:
+            config_dict = json.load(f)
+        
+        return cls(**{k: v for k, v in config_dict.items() 
+                     if k in cls.__dataclass_fields__ and k != "world"})