fix

modeling_spear.py

@@ -1,8 +1,8 @@
 # modeling_spear.py
 
 from transformers import PreTrainedModel
-from configuration_spear import SpearConfig
-from spear_model import SpearModel as model
+from .configuration_spear import SpearConfig
+from .spear_model import SpearModel as model
 
 
 class SpearModel(PreTrainedModel):

spear_model.py

@@ -26,8 +26,8 @@ import torch.nn as nn
 from torch.nn.utils.rnn import pad_sequence
 from torchaudio.compliance.kaldi import fbank as torch_fbank
 
-from configuration_spear import SpearConfig
-from zipformer import Zipformer2, Conv2dSubsampling
+from .configuration_spear import SpearConfig
+from .zipformer import Zipformer2, Conv2dSubsampling
 
 LOG_EPS=math.log(1e-10)
 SAMPLING_RATE=16000

zipformer.py

@@ -23,7 +23,7 @@ from typing import List, Optional, Tuple, Union
 import logging
 import torch
 import random
-from spear_modules import (
+from .spear_modules import (
     Balancer,
     BiasNorm,
     Dropout2,