Upload model_index.json with huggingface_hub

model_index.json

@@ -1,97 +1,36 @@
 {
+  "_class_name": "NexusWorldModel",
   "model_type": "nexus-worldmodel",
   "version": "2.0",
-  "architecture_type": "cognitive-dynamic",
-  "framework": "pytorch",
-  "library_name": "nexus-worldmodel",
-  
-  "description": "NEXUS-WorldModel v2.0 - Neural World Simulator with Full Cognitive Architecture",
-  
-  "features": [
-    "EARCP (Ensemble Auto-Regulated Coherence Protocol)",
-    "LPOL Memory with 9 world-specific domains",
-    "GQA (Grouped Query Attention) for efficient memory access",
-    "Neurogenesis (dynamic neuron growth)",
-    "Energy System for cognitive resource management",
-    "Dream Phase with prioritized replay",
-    "Multi-World Buffers (physical, social, abstract, temporal)",
-    "Physics Prior (Mixture Density Network)",
-    "RoPE Attention (Rotary Positional Encoding)",
-    "VAE for visual encoding/decoding"
-  ],
-  
-  "files": {
-    "config": "config.json",
-    "weights": "pytorch_model.bin",
-    "cognitive_state": "cognitive_state.json",
-    "training_state": "training_state.json"
-  },
-  
-  "optional_files": {
-    "optimizer": "optimizer.pt",
-    "source_code": "NEXUS_WorldModel_v2.py",
-    "config_module": "nexus_worldmodel_config.py"
-  },
-  
-  "load_instructions": {
-    "python": "from nexus_worldmodel_config import load_nexus_worldmodel\nfrom NEXUS_WorldModel_v2 import NexusWorldModel\n\nmodel, config, cognitive_state, warnings = load_nexus_worldmodel(\n    NexusWorldModel,\n    'amewebstudio/nexus-worldmodel-v2',\n    device='cuda'\n)",
-    "note": "Use strict=False to handle dynamic architecture size mismatches"
-  },
-  
-  "dynamic_components": {
-    "experts": {
-      "description": "Expert count per EARCP layer can grow during training",
-      "initial": 6,
-      "max": 12,
-      "growth_trigger": "coherence < 0.3"
+  "components": {
+    "vae_encoder": {
+      "latent_dim": 256
     },
-    "neurons": {
-      "description": "Neuron count in neurogenesis layer can change",
-      "initial": 64,
-      "min": 32,
-      "max": 256,
-      "birth_trigger": "coherence > 0.8",
-      "death_trigger": "usage < 0.05"
-    }
-  },
-  
-  "cognitive_systems": {
-    "energy": {
-      "description": "Manages cognitive resource consumption",
-      "think_cost": 0.02,
-      "dream_cost": 0.1,
-      "regeneration": 0.05
+    "vae_decoder": {
+      "latent_dim": 256
     },
-    "dream": {
-      "description": "Memory consolidation through prioritized replay",
-      "cycle_length": 50,
-      "duration": 10
+    "lpol_memory": {
+      "domains": 9,
+      "use_gqa": true
     },
-    "memory": {
-      "lpol_domains": 9,
-      "episodic_slots": 256,
-      "multi_scale": true
+    "gqa": {
+      "num_heads": 8,
+      "num_kv_groups": 2
+    },
+    "earcp_layers": {
+      "num_layers": 8,
+      "experts": 6
+    },
+    "neurogenesis": {
+      "enabled": true
+    },
+    "physics_prior": {
+      "components": 8
     }
   },
-  
   "training_info": {
-    "recommended_epochs": 15,
-    "recommended_batch_size": 32,
-    "learning_rate": 0.0001,
-    "optimizer": "AdamW",
-    "scheduler": "CosineAnnealingLR",
-    "mixed_precision": true
-  },
-  
-  "author": {
-    "name": "Mike Amega (Logo)",
-    "organization": "Ame Web Studio",
-    "email": "contact@amewebstudio.com"
-  },
-  
-  "license": "Apache-2.0",
-  
-  "citation": {
-    "bibtex": "@software{nexus_worldmodel_2025,\n  author = {Amega, Mike},\n  title = {NEXUS-WorldModel: Neural World Simulator with Cognitive Architecture},\n  year = {2025},\n  url = {https://huggingface.co/amewebstudio/nexus-worldmodel-v2}\n}"
+    "epochs": 6,
+    "loss": 0.01724659317859717,
+    "neurogenesis": 0
   }
-}
+}