[Bug] model_type argument as str for checkpoints classes

smujjiga commented 3 weeks ago

What is the purpose of this PR? Is it to

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

The correct usage of model_type argument is as a string as shown here

pytorch-bot[bot] commented 3 weeks ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchtune/1946

:page_facing_up: Preview Python docs built from this PR

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

:white_check_mark: No Failures

As of commit ca6f70720b40abd41e9d44f4cb25aed68887ef01 with merge base e861954d43d23f03f2857b0556916a8431766d5d (): :green_heart: Looks good so far! There are no failures yet. :green_heart:

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

ebsmothers commented 3 weeks ago

Thanks @smujjiga for the fix! This looks good to me, once CI is green we can merge

pytorch / torchtune