We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
dataset_name
create_model_card
1 parent 4845fd1 commit 9ae9994Copy full SHA for 9ae9994
hub_utils.py
@@ -168,13 +168,13 @@ def create_model_card(args, model_name):
168
license="apache-2.0",
169
library_name="diffusers",
170
tags=[],
171
- datasets=args.dataset,
+ datasets=args.dataset_name,
172
metrics=[],
173
),
174
template_path=MODEL_CARD_TEMPLATE_PATH,
175
model_name=model_name,
176
repo_name=repo_name,
177
- dataset_name=args.dataset if hasattr(args, "dataset") else None,
+ dataset_name=args.dataset_name if hasattr(args, "dataset_name") else None,
178
learning_rate=args.learning_rate,
179
train_batch_size=args.train_batch_size,
180
eval_batch_size=args.eval_batch_size,
0 commit comments