torchtune
torchtune copied to clipboard
Shrink `small_model` ckpt to run in pytest and be available under `assets/` dir
Currently, a user needs access to our Meta-enabled S3 bucket in order to run the full suite of tests. Any user who wants to contribute should be able to reason about the quality of their code. Therefore, we should shrink the size of our small model checkpoint in order to run through pytest. See an example for this in #271 .
@joecummings agree with this and assigning to you.
This is done.