fixes #2

model.py

@@ -1,11 +1,6 @@
 from transformers.models.whisper.configuration_whisper import WhisperConfig
-import torch.nn.functional as F
-from transformers.generation.logits_process import (
-    LogitsProcessorList,
-    SuppressTokensLogitsProcessor
-)
+
 from typing import List, Optional, Dict, Any
-import warnings
 
 """Custom config to support modification of the Whisper encoder."""
     
@@ -63,11 +58,18 @@ class CustomWhisperConfig(WhisperConfig):
 import math
 from typing import Optional, Tuple, Union
 
+import warnings
 import numpy as np
 import torch
 import torch.utils.checkpoint
 from torch import nn
 from torch.nn import CrossEntropyLoss
+import torch.nn.functional as F
+
+from transformers.generation.logits_process import (
+    LogitsProcessorList,
+    SuppressTokensLogitsProcessor
+)
 
 from transformers.activations import ACT2FN
 from transformers.cache_utils import Cache, DynamicCache, EncoderDecoderCache, StaticCache