teticio commited on
Commit
e0ef439
·
1 Parent(s): f8a240c

remove unnecessary imports

Browse files
Files changed (1) hide show
  1. scripts/train_unconditional.py +0 -4
scripts/train_unconditional.py CHANGED
@@ -11,16 +11,12 @@ from accelerate.logging import get_logger
11
  from datasets import load_from_disk, load_dataset
12
  from diffusers import (DiffusionPipeline, DDPMScheduler, UNet2DModel,
13
  DDIMScheduler, AutoencoderKL)
14
- from diffusers.modeling_utils import EntryNotFoundError
15
  from diffusers.hub_utils import init_git_repo, push_to_hub
16
  from diffusers.optimization import get_scheduler
17
  from diffusers.training_utils import EMAModel
18
  from torchvision.transforms import (
19
- CenterCrop,
20
  Compose,
21
- InterpolationMode,
22
  Normalize,
23
- Resize,
24
  ToTensor,
25
  )
26
  import numpy as np
 
11
  from datasets import load_from_disk, load_dataset
12
  from diffusers import (DiffusionPipeline, DDPMScheduler, UNet2DModel,
13
  DDIMScheduler, AutoencoderKL)
 
14
  from diffusers.hub_utils import init_git_repo, push_to_hub
15
  from diffusers.optimization import get_scheduler
16
  from diffusers.training_utils import EMAModel
17
  from torchvision.transforms import (
 
18
  Compose,
 
19
  Normalize,
 
20
  ToTensor,
21
  )
22
  import numpy as np