From eef3fb28fa79bff0b5d429ca7754c909a81ec62d Mon Sep 17 00:00:00 2001 From: teticio Date: Thu, 8 Dec 2022 21:17:40 +0000 Subject: [PATCH] add missing pathlib import --- scripts/train_unconditional.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/scripts/train_unconditional.py b/scripts/train_unconditional.py index 52e322b..5048d6f 100644 --- a/scripts/train_unconditional.py +++ b/scripts/train_unconditional.py @@ -2,11 +2,9 @@ import argparse import os +from pathlib import Path from typing import Optional -import torch -import torch.nn.functional as F - from accelerate import Accelerator from accelerate.logging import get_logger from datasets import load_from_disk, load_dataset @@ -18,17 +16,19 @@ AutoencoderKL, ) from diffusers.pipelines.audio_diffusion import Mel -from huggingface_hub import HfFolder, Repository, whoami from diffusers.optimization import get_scheduler from diffusers.training_utils import EMAModel +from huggingface_hub import HfFolder, Repository, whoami +from librosa.util import normalize +import numpy as np +import torch +import torch.nn.functional as F from torchvision.transforms import ( Compose, Normalize, ToTensor, ) -import numpy as np from tqdm.auto import tqdm -from librosa.util import normalize logger = get_logger(__name__)