Spaces:
Runtime error
Runtime error
remove unnecessary imports
Browse files
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
|