Handle tuple return from generate_dataset_group_by_blueprint

This commit is contained in:
rockerBOO
2025-01-08 18:38:20 -05:00
parent 556f3f1696
commit 9fde0d7972
15 changed files with 24 additions and 17 deletions

View File

@@ -467,7 +467,7 @@ class BlueprintGenerator:
return default_value
def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlueprint):
def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlueprint) -> Tuple[DatasetGroup, Optional[DatasetGroup]]:
datasets: List[Union[DreamBoothDataset, FineTuningDataset, ControlNetDataset]] = []
for dataset_blueprint in dataset_group_blueprint.datasets: