Upload folder using huggingface_hub

modeling_taonet.py

@@ -4,8 +4,8 @@ from torch import nn
 from transformers import GenerationMixin, PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutput
 
-from configuration_taonet import TaoNetConfig
-from taonet_model import SimpleLLM, build_runtime_config
+from .configuration_taonet import TaoNetConfig
+from .taonet_model import SimpleLLM, build_runtime_config
 
 
 class TaoNetForCausalLM(PreTrainedModel, GenerationMixin):

taonet_model.py

@@ -6,8 +6,8 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
-from embeddings import FactorizedEmbedding
-from mla_components import AttentionBlock
+from .embeddings import FactorizedEmbedding
+from .mla_components import AttentionBlock
 
 
 class SimpleLLM(nn.Module):