Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 20:19:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

custom diffusion

Browse Source
This commit is contained in:
XDUWQ
2023-07-12 16:00:18 +08:00
parent 73fe5d5d65
commit d18cf31dfa
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/trainers/multi_modal/custom_diffusion/custom_diffusion_trainer.py
View File

@@ -5,7 +5,6 @@ import os
import random
import shutil
import warnings
from collections.abc import Mapping
from pathlib import Path
from typing import Union
@@ -23,7 +22,7 @@ from tqdm.auto import tqdm
from modelscope.metainfo import Trainers
from modelscope.msdatasets import MsDataset
from modelscope.outputs import ModelOutputBase, OutputKeys
from modelscope.outputs import OutputKeys
from modelscope.trainers.builder import TRAINERS
from modelscope.trainers.hooks.checkpoint.checkpoint_hook import CheckpointHook
from modelscope.trainers.hooks.checkpoint.checkpoint_processor import \
@@ -33,7 +32,6 @@ from modelscope.trainers.trainer import EpochBasedTrainer
from modelscope.utils.config import ConfigDict
from modelscope.utils.constant import ModeKeys, TrainerStages
from modelscope.utils.data_utils import to_device
from modelscope.utils.file_utils import func_receive_dict_inputs
from modelscope.utils.torch_utils import is_dist