Fix imports for trust_remote_code: latent_thought.py
Browse files- 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)
|