fix: setting log level in save()

foundation-model-stack / fms-hf-tuning

🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.

Apache License 2.0

28 stars 48 forks source link

fix: setting log level in save() #304

Closed anhuong closed 3 months ago

anhuong commented 3 months ago

Description of the change

Bug where log level set in save() fails if lowercase values passed in for log_level. This fixes the issue and adds a small addition to unit tests to ensure this change fixes it.

Related issue number

How to verify the PR

Was the PR tested

[x] I have added >=1 unit test(s) for every new method I have added.
[ ] I have ensured all unit tests pass

Abhishek-TAMU commented 3 months ago

Looks good to me!