style

dalle_mini/model/modeling.py

@@ -30,21 +30,20 @@ from transformers.modeling_flax_outputs import (
     FlaxSeq2SeqLMOutput,
 )
 from transformers.modeling_flax_utils import ACT2FN
-from transformers.utils import logging
-
 from transformers.models.bart.modeling_flax_bart import (
     FlaxBartAttention,
-    FlaxBartEncoderLayer,
+    FlaxBartDecoder,
     FlaxBartDecoderLayer,
-    FlaxBartEncoderLayerCollection,
     FlaxBartDecoderLayerCollection,
     FlaxBartEncoder,
-    FlaxBartDecoder,
-    FlaxBartModule,
+    FlaxBartEncoderLayer,
+    FlaxBartEncoderLayerCollection,
+    FlaxBartForConditionalGeneration,
     FlaxBartForConditionalGenerationModule,
+    FlaxBartModule,
     FlaxBartPreTrainedModel,
-    FlaxBartForConditionalGeneration,
 )
+from transformers.utils import logging
 
 from .configuration import DalleBartConfig
 

tools/train/train.py

@@ -43,7 +43,7 @@ from tqdm import tqdm
 from transformers import AutoTokenizer, HfArgumentParser
 
 from dalle_mini.data import Dataset
-from dalle_mini.model import DalleBartConfig, DalleBart
+from dalle_mini.model import DalleBart, DalleBartConfig
 
 logger = logging.getLogger(__name__)