Update modeling_mistral.py

modeling_mistral.py

@@ -1,4 +1,3 @@
-import logging
 from typing import Optional, List, Tuple, Union
 
 import torch
@@ -9,6 +8,7 @@ from transformers.modeling_attn_mask_utils import _prepare_4d_causal_attention_m
 from transformers.modeling_outputs import BaseModelOutputWithPast
 from transformers.models.mistral.modeling_mistral import MistralPreTrainedModel, MistralDecoderLayer, MistralRMSNorm, \
     MistralForCausalLM
+from transformers.utils import logging
 from .configuration_mistral import MistralDenseFormerConfig
 from .denseformer import DWAModules