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

custom diffusion

Browse Source
This commit is contained in:
XDUWQ
2023-07-12 17:15:10 +08:00
parent d18cf31dfa
commit c1aa120029
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,7 +3,6 @@ import hashlib
import itertools
import os
import random
import shutil
import warnings
from pathlib import Path
from typing import Union
@@ -296,8 +295,8 @@ class CustomDiffusionTrainer(EpochBasedTrainer):
if os.path.isdir(instance_data_name):
self.instance_data_dir = instance_data_name
else:
ds = MsDataset.load(instance_data_name)
self.instance_data_dir = next(iter(ds))[-1] + '/data'
ds = MsDataset.load(instance_data_name, split='train')
self.instance_data_dir = os.path.dirname(next(iter(ds))['Target:FILE'])
self.concepts_list = [{
'instance_prompt': self.instance_prompt,