Update memory.py

memory.py

@@ -17,9 +17,9 @@ class CognitiveMemory(nn.Module):
         self.consolidation_threshold = 0.7
         
         # Memory projection layers
-        self.key_proj = nn.Linear(1, 64)  # Changed from context_size to 1
-        self.value_proj = nn.Linear(1, 64)  # Changed from context_size to 1
-        
+        self.key_proj = nn.Linear(context_size, 64)
+        self.value_proj = nn.Linear(context_size, 64)
+
     def add_memory(self, context: torch.Tensor, activation: float):
         """Store new memory with adaptive importance"""
         # Ensure context is 1D tensor with single value