Skip to content

GitLab

Unverified Commit 70d09854 authored by AisingioroHao's avatar AisingioroHao Committed by GitHub
Browse files

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


Co-authored-by: default avatarAisingioroHao0 <1286098622@qq.com>
parent ea1fcc28
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/text_to_image/train_text_to_image.py
View file @ 70d09854
...@@ -716,6 +716,7 @@ def main(): ...@@ -716,6 +716,7 @@ def main():
args.dataset_name, args.dataset_name,
args.dataset_config_name, args.dataset_config_name,
cache_dir=args.cache_dir, cache_dir=args.cache_dir,
data_dir=args.train_data_dir,
) )
else: else:
data_files = {} data_files = {}
......
examples/text_to_image/train_text_to_image_lora.py
View file @ 70d09854
...@@ -559,6 +559,7 @@ def main(): ...@@ -559,6 +559,7 @@ def main():
args.dataset_name, args.dataset_name,
args.dataset_config_name, args.dataset_config_name,
cache_dir=args.cache_dir, cache_dir=args.cache_dir,
data_dir=args.train_data_dir,
) )
else: else:
data_files = {} data_files = {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment