hell2.0

configuration_super_linear.py

@@ -1,6 +1,6 @@
 from typing import Optional, Tuple
 import torch, torch.nn as nn, torch.nn.functional as F
-from configuration_super_linear_base import SuperLinearConfigBase
+from .configuration_super_linear_base import SuperLinearConfigBase
 
 
 # 1) --------------------------------------------------------------------------

configuration_super_linear_fs.py

@@ -1,7 +1,7 @@
 from typing import Optional, Tuple
 import torch, torch.nn as nn, torch.nn.functional as F
 
-from configuration_super_linear_base import SuperLinearConfigBase
+from .configuration_super_linear_base import SuperLinearConfigBase
 from transformers.modeling_outputs import CausalLMOutputWithCrossAttentions
 
 # 1) --------------------------------------------------------------------------

modeling_super_linear.py

@@ -7,6 +7,7 @@ from transformers.modeling_outputs         import CausalLMOutputWithCrossAttenti
 from .configuration_super_linear_base      import SuperLinearConfigBase
 from .configuration_super_linear           import SuperLinearConfig
 from .configuration_super_linear_fs        import SuperLinearConfigFS
+
 from typing import Tuple, Union