add load_datasete data_dir parameter (huggingface#5747)

Skquark · Nov 10, 2023 · 1f87f83 · 1f87f83
1 parent 77ba494
commit 1f87f83
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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 = {}