Kiy-K commited on
Commit
4fcbdfa
·
verified ·
1 Parent(s): 6c8b26c

Update modeling_kiyengine.py

Browse files
Files changed (1) hide show
  1. modeling_kiyengine.py +1 -1
modeling_kiyengine.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import torch.nn as nn
7
  from transformers import PreTrainedModel
8
  from transformers.modeling_outputs import BaseModelOutput
9
- from configuration_kiyengine import KiyEngineConfig
10
 
11
 
12
  class MambaBlock(nn.Module):
 
6
  import torch.nn as nn
7
  from transformers import PreTrainedModel
8
  from transformers.modeling_outputs import BaseModelOutput
9
+ from .configuration_kiyengine import KiyEngineConfig
10
 
11
 
12
  class MambaBlock(nn.Module):