Commit ef504c78 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent a062e47e
...@@ -24,7 +24,6 @@ from flax.jax_utils import unreplicate ...@@ -24,7 +24,6 @@ from flax.jax_utils import unreplicate
from flax.training.common_utils import shard from flax.training.common_utils import shard
from packaging import version from packaging import version
from PIL import Image from PIL import Image
from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel
from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel
......
...@@ -23,7 +23,6 @@ from flax.core.frozen_dict import FrozenDict ...@@ -23,7 +23,6 @@ from flax.core.frozen_dict import FrozenDict
from flax.jax_utils import unreplicate from flax.jax_utils import unreplicate
from flax.training.common_utils import shard from flax.training.common_utils import shard
from PIL import Image from PIL import Image
from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel
from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel
......
...@@ -24,7 +24,6 @@ from flax.jax_utils import unreplicate ...@@ -24,7 +24,6 @@ from flax.jax_utils import unreplicate
from flax.training.common_utils import shard from flax.training.common_utils import shard
from packaging import version from packaging import version
from PIL import Image from PIL import Image
from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel from transformers import CLIPFeatureExtractor, CLIPTokenizer, FlaxCLIPTextModel
from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment