Add a data_dir parameter to the load_dataset method. (#4482)

Co-authored-by: AisingioroHao0 <1286098622@qq.com>

examples/text_to_image/train_text_to_image.py

@@ -716,6 +716,7 @@ def main():
             args.dataset_name,
             args.dataset_config_name,
             cache_dir=args.cache_dir,
+            data_dir=args.train_data_dir,
         )
     else:
         data_files = {}

examples/text_to_image/train_text_to_image_lora.py

@@ -559,6 +559,7 @@ def main():
             args.dataset_name,
             args.dataset_config_name,
             cache_dir=args.cache_dir,
+            data_dir=args.train_data_dir,
         )
     else:
         data_files = {}