Skip to content

Commit 70d0985

Browse files
aihao2000AisingioroHao0
aihao2000
and
AisingioroHao0
authored
Add a data_dir parameter to the load_dataset method. (huggingface#4482)
Co-authored-by: AisingioroHao0 <1286098622@qq.com>
1 parent ea1fcc2 commit 70d0985

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

‎examples/text_to_image/train_text_to_image.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -716,6 +716,7 @@ def load_model_hook(models, input_dir):
716716
args.dataset_name,
717717
args.dataset_config_name,
718718
cache_dir=args.cache_dir,
719+
data_dir=args.train_data_dir,
719720
)
720721
else:
721722
data_files = {}

‎examples/text_to_image/train_text_to_image_lora.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -559,6 +559,7 @@ def compute_snr(timesteps):
559559
args.dataset_name,
560560
args.dataset_config_name,
561561
cache_dir=args.cache_dir,
562+
data_dir=args.train_data_dir,
562563
)
563564
else:
564565
data_files = {}

0 commit comments

Comments
 (0)