Update modeling_mamba.py
Browse files- 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
         
     |