fix missing key

[felipemello1]

Context

What is the purpose of this PR? Is it to

• add a new feature
• fix a bug
• update tests and/or documentation
• other (please add here)

"None" was missing in a couple of places.

without it:

with it:

tune run lora_finetune_single_device --config llama3/8B_dora_single_device \
optimizer_in_bwd=True \
enable_activation_checkpointing=True \
enable_activation_offloading=True \
gradient_accumulation_steps=1 \
compile=True \
dataset.packed=True \
dataset.split=train[:10%] \
tokenizer.max_seq_len=2048 \
metric_logger=torchtune.training.metric_logging.WandBLogger \
metric_logger.project=profiling \
metric_logger.tags=[my_experiment_name] \
log_every_n_steps=1 \
log_peak_memory_stats=True \
max_steps_per_epoch=15 \
epochs=2 \
batch_size=6 \
metric_logger.name=Dora_Save_adapter_weights_only \
save_adapter_weights_only=True

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchtune/1952

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit c38a46d with merge base 24d3579 ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.