Formatting

mosaicml · Nov 8, 2024 · ecf1c2f · ecf1c2f
1 parent 68e6a5f
commit ecf1c2f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/diffusion/models/models.py b/diffusion/models/models.py
@@ -9,7 +9,8 @@
 
 import torch
 from composer.devices import DeviceGPU
-from diffusers import AutoencoderKL, DDIMScheduler, DDPMScheduler, DPMSolverMultistepScheduler, EulerDiscreteScheduler, UNet2DConditionModel
+from diffusers import (AutoencoderKL, DDIMScheduler, DDPMScheduler, DPMSolverMultistepScheduler, EulerDiscreteScheduler,
+                       UNet2DConditionModel)
 from peft import LoraConfig
 from torchmetrics import MeanSquaredError
 from transformers import AutoModel, AutoTokenizer, CLIPTextModel, CLIPTokenizer, PretrainedConfig