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
|