Skip to content

Commit

Permalink
Fix arg key for dataset_name in create_model_card (open-mmlab#158)
Browse files Browse the repository at this point in the history
Fix arg key for `dataset_name`

The example training script was changed in open-mmlab#152, but not
`create_model_card`.
  • Loading branch information
pcuenca authored Aug 6, 2022
1 parent a209037 commit c4ae7c2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/diffusers/hub_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -168,13 +168,13 @@ def create_model_card(args, model_name):
license="apache-2.0",
library_name="diffusers",
tags=[],
datasets=args.dataset,
datasets=args.dataset_name,
metrics=[],
),
template_path=MODEL_CARD_TEMPLATE_PATH,
model_name=model_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,
train_batch_size=args.train_batch_size,
eval_batch_size=args.eval_batch_size,
Expand Down

0 comments on commit c4ae7c2

Please sign in to comment.