From 5470bb34b9385c0a5559f9d91ed937cdbf7d2296 Mon Sep 17 00:00:00 2001 From: aihao2000 Date: Fri, 10 Nov 2023 16:42:39 +0800 Subject: [PATCH] add load_datasete data_dir parameter --- examples/text_to_image/train_text_to_image_flax.py | 1 + examples/text_to_image/train_text_to_image_lora_sdxl.py | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/text_to_image/train_text_to_image_flax.py b/examples/text_to_image/train_text_to_image_flax.py index 64b71b4f83ae..e5410a0a9e90 100644 --- a/examples/text_to_image/train_text_to_image_flax.py +++ b/examples/text_to_image/train_text_to_image_flax.py @@ -275,6 +275,7 @@ def main(): args.dataset_name, args.dataset_config_name, cache_dir=args.cache_dir, + data_dir=args.train_data_dir ) else: data_files = {} diff --git a/examples/text_to_image/train_text_to_image_lora_sdxl.py b/examples/text_to_image/train_text_to_image_lora_sdxl.py index f0d83d55e9bf..d621858e3d2c 100644 --- a/examples/text_to_image/train_text_to_image_lora_sdxl.py +++ b/examples/text_to_image/train_text_to_image_lora_sdxl.py @@ -768,6 +768,7 @@ def load_model_hook(models, input_dir): args.dataset_name, args.dataset_config_name, cache_dir=args.cache_dir, + data_dir=args.train_data_dir ) else: data_files = {}