Unverified Commit c4ae7c24 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix arg key for `dataset_name` in `create_model_card` (#158)

Fix arg key for `dataset_name`

The example training script was changed in #152, but not
`create_model_card`.
parent a2090375
...@@ -168,13 +168,13 @@ def create_model_card(args, model_name): ...@@ -168,13 +168,13 @@ def create_model_card(args, model_name):
license="apache-2.0", license="apache-2.0",
library_name="diffusers", library_name="diffusers",
tags=[], tags=[],
datasets=args.dataset, datasets=args.dataset_name,
metrics=[], metrics=[],
), ),
template_path=MODEL_CARD_TEMPLATE_PATH, template_path=MODEL_CARD_TEMPLATE_PATH,
model_name=model_name, model_name=model_name,
repo_name=repo_name, repo_name=repo_name,
dataset_name=args.dataset if hasattr(args, "dataset") else None, dataset_name=args.dataset_name if hasattr(args, "dataset_name") else None,
learning_rate=args.learning_rate, learning_rate=args.learning_rate,
train_batch_size=args.train_batch_size, train_batch_size=args.train_batch_size,
eval_batch_size=args.eval_batch_size, eval_batch_size=args.eval_batch_size,
......
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