remove unused imports

Signed-off-by: Xuesong Yang <[email protected]>

nemo/collections/nlp/modules/common/megatron/attention.py

@@ -39,7 +39,6 @@
     attention_mask_func,
 )
 from nemo.core import adapter_mixins
-from nemo.utils import logging
 
 try:
     from apex.transformer.enums import AttnMaskType, AttnType

nemo/collections/nlp/modules/common/megatron/megatron_decoders.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 """Transformer based language model."""
-from ast import Mod
 from nemo.collections.nlp.modules.common.megatron.layer_type import LayerType
 from nemo.collections.nlp.modules.common.megatron.megatron_transformer_decoder import MegatronTransformerDecoderModule
 from nemo.collections.nlp.modules.common.megatron.retrieval_transformer import (

nemo/collections/nlp/modules/common/megatron/megatron_encoder_decoder.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 """Transformer based language model."""
-from ast import Mod
 
 import torch
 

nemo/collections/nlp/modules/common/megatron/utils.py

@@ -18,7 +18,6 @@
 from typing import Dict, Iterator, List, Optional, Tuple, Union
 
 import torch
-import torch.nn as nn
 from torch import Tensor
 
 from nemo.utils import logging, logging_mode

nemo/collections/tts/models/speechllm/megatron_t5_speechllm_model.py

@@ -76,8 +76,7 @@
 import time
 
 import librosa
-import torchaudio
-from torchaudio.pipelines import SQUIM_OBJECTIVE, SQUIM_SUBJECTIVE
+from torchaudio.pipelines import SQUIM_SUBJECTIVE
 from transformers import Wav2Vec2FeatureExtractor, WavLMForXVector
 
 __all__ = ['MegatronT5SpeechLMModel']

nemo/collections/tts/parts/utils/helpers.py

@@ -50,7 +50,6 @@
 import numpy as np
 import seaborn as sns
 import torch
-from einops import rearrange
 from numba import jit, prange
 
 from nemo.collections.tts.torch.tts_data_types import DATA_STR2DATA_CLASS, MAIN_DATA_TYPES, WithLens