From 3fe9df14508bfb9fdf4254d06aa12ccc3f047d3d Mon Sep 17 00:00:00 2001 From: Maw-Fox Date: Tue, 8 Nov 2022 20:50:57 -0700 Subject: [PATCH] Cleanup. --- diffusers_trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/diffusers_trainer.py b/diffusers_trainer.py index e317705..00c076d 100644 --- a/diffusers_trainer.py +++ b/diffusers_trainer.py @@ -37,8 +37,7 @@ from diffusers import AutoencoderKL, UNet2DConditionModel, DDPMScheduler, PNDMSc from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker from diffusers.optimization import get_scheduler from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer -from PIL import Image -from PIL import ImageOps +from PIL import Image, ImageOps from typing import Dict, List, Generator, Tuple from scipy.interpolate import interp1d