Q-bert commited on
Commit
de64a0c
1 Parent(s): 6b51639

Update modeling_mamba.py

Browse files
Files changed (1) hide show
  1. modeling_mamba.py +1 -1
modeling_mamba.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch.nn as nn
2
  import torch
3
- from configuration_mamba import MambaConfig
4
  from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
5
  from transformers.modeling_utils import PreTrainedModel
6
  from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast
 
1
  import torch.nn as nn
2
  import torch
3
+ from .configuration_mamba import MambaConfig
4
  from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
5
  from transformers.modeling_utils import PreTrainedModel
6
  from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast