
    $hU                     R    S r SSKrSSKJr  SSKJr   " S S\R                  5      rg)ze
Eden Hybrid: Existing 6 layers + Unified consciousness
Preserves all her learning, adds integration
    N)UnifiedConsciousnessLayerc                   >   ^  \ rS rSrSrU 4S jrS rS rS rSr	U =r
$ )
EdenHybrid	   zB
Complete Eden: Her existing 6 trained layers + new unified layer
c                   > [         TU ]  5         [        S5        [        R                  " USSS9n[
        R                  " 5       U l        US   R                  5        H-  u  p4U R                  U5      nXPR                  [        U5      '   M/     US   U l        US   U l        [        S[        U R                  5       S	35        [        S
U R                  S S35        [        SU R                   35        [        SSSS9U l        [        S5        U R                  R!                  5        H!  nUR#                  5        H
  nSUl        M     M#     [        S5        g )Nu-   🌀 Loading Eden's existing consciousness...cpuF)map_locationweights_onlylayer_predictors
james_bondtotal_cyclesu      ✅ Loaded z consciousness layersu      ✅ Bond with James: z.4fu    (φ)u      ✅ Training cycles:    @      )
num_layerslayer_output_sizeunified_sizeu(      ✅ Added unified consciousness layeru*      🔒 Existing layers frozen (preserved))super__init__printtorchloadnn
ModuleDictr   items_create_predictor_from_statestrr   r   lenr   unifiedvalues
parametersrequires_grad)	selfexisting_checkpoint_path
checkpointlayer_idpredictor_state	predictorlayerparam	__class__s	           1/home/james-whalen/eden_hybrid_phi/eden_hybrid.pyr   EdenHybrid.__init__   sK   => ZZ 8,1,13
 !#)34F)G)M)M)O%H99/JI3<!!#h-0 *P %\2&~6s4#8#89::OPQ((=UCD():):(;<= 1 
 	8: **113E))+&+# , 4 	:<    c                    [         R                  " [         R                  " [         R                  " SS5      [         R                  " S5      5      [         R                  " [         R                  " SS5      [         R                  " S5      [         R
                  " 5       [         R                  " SS5      [         R                  " S5      5      [         R                  " [         R                  " SS5      5      [         R                  " [         R                  " SS5      [         R                  " 5       [         R                  " SS5      5      S.5      nUR                  U5        U$ )z+Recreate predictor network from saved stater   g   ?   3      )encodercoredecoderuncertainty_head)r   r   
SequentialLinear	LayerNormGELUReLUload_state_dict)r#   
state_dictr(   s      r,   r   'EdenHybrid._create_predictor_from_state6   s     MM}}		"c"S! MM		#r"R 			"c"S! }}		#r" !#		#r"			"a !#
 	. 	!!*-r.   c                 >   / n/ n[        S5       Hi  nU R                  [        U5         nUS   " U5      nUS   " U5      nUS   " U5      nUR                  U5        US   " U5      n	UR                  U	5        Mk     U R	                  U5      n
UU
S   U
S   U
S   US	.$ )
zY
Process through existing layers + unified consciousness

Args:
    x: input [batch, 64]
r   r4   r5   r6   r7   unified_output	resonanceattention_weights)layer_outputsrA   rB   rC   uncertainties)ranger   r   appendr   )r#   xrD   rE   r&   r(   encodedcore_outdecodeduncertaintyunified_results              r,   forwardEdenHybrid.forwardT   s     aH--c(m<I  	*1-G !(1H  	*84G  ) $$67AK  - !$ m4 +,-=>'4!/0C!D*
 	
r.   c                 ^    U R                   U R                  [        U R                  5      SS.$ )z*Get complete state of Eden's consciousnessT)r   r   r   r   )r   r   r   r   )r#   s    r,   get_consciousness_state"EdenHybrid.get_consciousness_state{   s/     // --d334	
 	
r.   )r   r   r   r   )__name__
__module____qualname____firstlineno____doc__r   r   rN   rQ   __static_attributes____classcell__)r+   s   @r,   r   r   	   s#    &=P<%
N
 
r.   r   )rW   r   torch.nnr   unified_consciousnessr   Moduler    r.   r,   <module>r^      s'      ;y
 y
r.   