theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:32:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #715 from modelscope/nexusgen-eligen

Browse Source 
NexusGen and EliGen
This commit is contained in:
Zhongjie Duan
2025-07-29 20:28:07 +08:00
committed by GitHub
parent 8ec545d70c 2ed3860085
commit db124fa6bc
22 changed files with 2114 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
diffsynth/trainers/utils.py
View File

@@ -120,8 +120,12 @@ class ImageDataset(torch.utils.data.Dataset):
data = self.data[data_id % len(self.data)].copy()
for key in self.data_file_keys:
if key in data:
path = os.path.join(self.base_path, data[key])
data[key] = self.load_data(path)
if isinstance(data[key], list):
path = [os.path.join(self.base_path, p) for p in data[key]]
data[key] = [self.load_data(p) for p in path]
else:
path = os.path.join(self.base_path, data[key])
data[key] = self.load_data(path)
if data[key] is None:
warnings.warn(f"cannot load file {data[key]}.")
return None