From e0f007f2a912ba4312c6e4d622667fc03a891058 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Tue, 14 Feb 2023 20:55:38 +0900 Subject: [PATCH] Fix import --- train_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_network.py b/train_network.py index fdc466ec..b783379b 100644 --- a/train_network.py +++ b/train_network.py @@ -1,6 +1,7 @@ from diffusers.optimization import SchedulerType, TYPE_TO_SCHEDULER_FUNCTION from torch.optim import Optimizer from torch.cuda.amp import autocast +from torch.nn.parallel import DistributedDataParallel as DDP from typing import Optional, Union import importlib import argparse