Be2Jay commited on
Commit
2fa4b94
·
verified ·
1 Parent(s): 2443f79

Fix imports for trust_remote_code: latent_thought.py

Browse files
Files changed (1) hide show
  1. latent_thought.py +2 -2
latent_thought.py CHANGED
@@ -10,7 +10,7 @@ import torch.nn as nn
10
  import torch.nn.functional as F
11
  from typing import Optional, Tuple
12
 
13
- from configuration_aether_micro import AETHERMicroConfig
14
 
15
 
16
  class AETHERMicroLatentThought(nn.Module):
@@ -233,7 +233,7 @@ def integrate_ltl_into_decoder_layer(decoder_layer, config: AETHERMicroConfig):
233
  Integrate LTL into existing AETHERMicroDecoderLayer
234
 
235
  Usage:
236
- from latent_thought import integrate_ltl_into_decoder_layer
237
 
238
  # After creating decoder layer
239
  decoder_layer = AETHERMicroDecoderLayer(config)
 
10
  import torch.nn.functional as F
11
  from typing import Optional, Tuple
12
 
13
+ from .configuration_aether_micro import AETHERMicroConfig
14
 
15
 
16
  class AETHERMicroLatentThought(nn.Module):
 
233
  Integrate LTL into existing AETHERMicroDecoderLayer
234
 
235
  Usage:
236
+ from .latent_thought import integrate_ltl_into_decoder_layer
237
 
238
  # After creating decoder layer
239
  decoder_layer = AETHERMicroDecoderLayer(config)