isort

pipeline.py

@@ -1,39 +1,26 @@
-import torch
-import torch.nn.functional as F
 import inspect
-import numpy as np
-from typing import Callable, List, Optional, Union, Any
-from transformers import (
-    CLIPTextModel,
-    CLIPTokenizer,
-    CLIPVisionModel,
-    CLIPImageProcessor,
-)
-from diffusers import AutoencoderKL, DiffusionPipeline
-from diffusers.utils import (
-    deprecate,
-    is_accelerate_available,
-    is_accelerate_version,
-    logging,
-)
-from diffusers.configuration_utils import FrozenDict
-from diffusers.schedulers import DDIMScheduler
-from diffusers.utils.torch_utils import randn_tensor
-
 import math
 from inspect import isfunction
+from typing import Any, Callable, List, Optional, Union
 
+import numpy as np
+import torch
 import torch.nn as nn
-from einops import rearrange, repeat
-
-from diffusers.configuration_utils import ConfigMixin
-from diffusers.models.modeling_utils import ModelMixin
-
+import torch.nn.functional as F
 # require xformers!
 import xformers
 import xformers.ops
-
+from diffusers import AutoencoderKL, DiffusionPipeline
+from diffusers.configuration_utils import ConfigMixin, FrozenDict
+from diffusers.models.modeling_utils import ModelMixin
+from diffusers.schedulers import DDIMScheduler
+from diffusers.utils import (deprecate, is_accelerate_available,
+                             is_accelerate_version, logging)
+from diffusers.utils.torch_utils import randn_tensor
+from einops import rearrange, repeat
 from kiui.cam import orbit_camera
+from transformers import (CLIPImageProcessor, CLIPTextModel, CLIPTokenizer,
+                          CLIPVisionModel)
 
 
 def get_camera(