Skip to content

Commit 9ae9994

Browse files
authored
Fix arg key for dataset_name in create_model_card (huggingface#158)
Fix arg key for `dataset_name` The example training script was changed in huggingface#152, but not `create_model_card`.
1 parent 4845fd1 commit 9ae9994

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hub_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -168,13 +168,13 @@ def create_model_card(args, model_name):
168168
license="apache-2.0",
169169
library_name="diffusers",
170170
tags=[],
171-
datasets=args.dataset,
171+
datasets=args.dataset_name,
172172
metrics=[],
173173
),
174174
template_path=MODEL_CARD_TEMPLATE_PATH,
175175
model_name=model_name,
176176
repo_name=repo_name,
177-
dataset_name=args.dataset if hasattr(args, "dataset") else None,
177+
dataset_name=args.dataset_name if hasattr(args, "dataset_name") else None,
178178
learning_rate=args.learning_rate,
179179
train_batch_size=args.train_batch_size,
180180
eval_batch_size=args.eval_batch_size,

0 commit comments

Comments
 (0)