Change import order again.

My version of `isort` has a different opinion than the one in the github
action.

src/dalle_mini/model/modeling.py

@@ -22,6 +22,7 @@ from typing import Optional, Tuple
 import flax.linen as nn
 import jax
 import jax.numpy as jnp
+import wandb
 from flax.core.frozen_dict import unfreeze
 from flax.linen import make_causal_mask
 from flax.traverse_util import flatten_dict
@@ -46,8 +47,6 @@ from transformers.models.bart.modeling_flax_bart import (
 )
 from transformers.utils import logging
 
-import wandb
-
 from .configuration import DalleBartConfig
 
 logger = logging.get_logger(__name__)