VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix Input type error when using DDP

Browse Source
This commit is contained in:
Isotr0py
2023-02-08 19:34:03 +08:00
parent 5e96e1369d
commit b599adc938
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train_network.py
View File

@@ -1,5 +1,6 @@
from diffusers.optimization import SchedulerType, TYPE_TO_SCHEDULER_FUNCTION from diffusers.optimization import SchedulerType, TYPE_TO_SCHEDULER_FUNCTION
from torch.optim import Optimizer from torch.optim import Optimizer
from torch.cuda.amp import autocast
from typing import Optional, Union from typing import Optional, Union
import importlib import importlib
import argparse import argparse